Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -32,8 +32,8 @@ model.eval()
|
|
32 |
@spaces.GPU(duration=90)
|
33 |
def generate(
|
34 |
message: str,
|
35 |
-
max_new_tokens: int =
|
36 |
-
temperature: float = 0
|
37 |
top_p: float = 0.9,
|
38 |
top_k: int = 50,
|
39 |
repetition_penalty: float = 1.2,
|
|
|
32 |
@spaces.GPU(duration=90)
|
33 |
def generate(
|
34 |
message: str,
|
35 |
+
max_new_tokens: int = 420,
|
36 |
+
temperature: float = 0,
|
37 |
top_p: float = 0.9,
|
38 |
top_k: int = 50,
|
39 |
repetition_penalty: float = 1.2,
|