Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ ai = aitextgen(model="chido/vggAI-offlinechat")
|
|
24 |
def get_text(user_input = '', temp=0.8):
|
25 |
#result = generator(user_input, temperature=temp)
|
26 |
#return result[0]["generated_text"]
|
27 |
-
return ai.generate_one(prompt=user_input, max_length=
|
28 |
|
29 |
gr.Interface(
|
30 |
get_text,
|
|
|
24 |
def get_text(user_input = '', temp=0.8):
|
25 |
#result = generator(user_input, temperature=temp)
|
26 |
#return result[0]["generated_text"]
|
27 |
+
return ai.generate_one(prompt=user_input, max_length=256, temperature=temp, top_p=0.8, repetition_penalty=1.25)
|
28 |
|
29 |
gr.Interface(
|
30 |
get_text,
|