diabolic6045 commited on
Commit
e166515
1 Parent(s): 8447407

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import os
7
  login(os.environ['HF_KEY'])
8
 
9
  # Define a function that takes user input and returns the model's output
10
- @spaces.GPU(duration=120)
11
  def generate_text(prompt):
12
  config = PeftConfig.from_pretrained("diabolic6045/gemma-2-2b-chess-adapter")
13
  base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b")
 
7
  login(os.environ['HF_KEY'])
8
 
9
  # Define a function that takes user input and returns the model's output
10
+ @spaces.GPU(duration=20)
11
  def generate_text(prompt):
12
  config = PeftConfig.from_pretrained("diabolic6045/gemma-2-2b-chess-adapter")
13
  base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b")