Spaces:
Sleeping
Sleeping
diabolic6045
commited on
Commit
•
e166515
1
Parent(s):
8447407
Update app.py
Browse files
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=
|
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")
|