chido commited on
Commit
71f2616
1 Parent(s): 4575d6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=400, temperature=temp, top_p=0.8, repetition_penalty=1.25)
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,