Spaces:

diabolic6045
/

gemma2-2b-chess

Running on Zero

diabolic6045 commited on 2 days ago

Commit

54dacf6

•

1 Parent(s): e166515

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import os
 login(os.environ['HF_KEY'])
 # Define a function that takes user input and returns the model's output
-@spaces.GPU(duration=20)
 def generate_text(prompt):
     config = PeftConfig.from_pretrained("diabolic6045/gemma-2-2b-chess-adapter")
     base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b")

 login(os.environ['HF_KEY'])
 # Define a function that takes user input and returns the model's output
+@spaces.GPU(duration=120)
 def generate_text(prompt):
     config = PeftConfig.from_pretrained("diabolic6045/gemma-2-2b-chess-adapter")
     base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b")