Spaces:
Runtime error
Runtime error
ai-forever
commited on
Commit
•
95590bf
1
Parent(s):
a9069b3
Update app.py
Browse files
app.py
CHANGED
@@ -32,8 +32,8 @@ def generate(prompt: str):
|
|
32 |
#length = int(length)
|
33 |
input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
|
34 |
out = model.generate(input_ids,
|
35 |
-
min_length=
|
36 |
-
max_length=
|
37 |
top_p=0.95,
|
38 |
top_k=0,
|
39 |
temperature=0.9,
|
@@ -58,4 +58,4 @@ interface = gr.Interface.load("huggingface/sberbank-ai/mGPT-armenian",
|
|
58 |
article = article
|
59 |
)
|
60 |
|
61 |
-
interface.launch(enable_queue=True)
|
|
|
32 |
#length = int(length)
|
33 |
input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
|
34 |
out = model.generate(input_ids,
|
35 |
+
min_length=200,
|
36 |
+
max_length=700,
|
37 |
top_p=0.95,
|
38 |
top_k=0,
|
39 |
temperature=0.9,
|
|
|
58 |
article = article
|
59 |
)
|
60 |
|
61 |
+
interface.launch(enable_queue=True)
|