MatteoScript commited on
Commit
46f724e
·
1 Parent(s): c3e5a9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -12,7 +12,7 @@ def format_prompt(message, history):
12
  return prompt
13
 
14
  def generate(
15
- prompt, history, temperature=0.2, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
16
  ):
17
  temperature = float(temperature)
18
  if temperature < 1e-2:
@@ -38,9 +38,6 @@ def generate(
38
  yield output
39
  return output
40
 
41
- iface = gr.Interface(fn=generate, inputs="text", outputs="text")
42
- iface.launch()
43
-
44
  mychatbot = gr.Chatbot(
45
  avatar_images=["./user.png", "./botm.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True,)
46
 
 
12
  return prompt
13
 
14
  def generate(
15
+ prompt, history, temperature=0.2, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0,
16
  ):
17
  temperature = float(temperature)
18
  if temperature < 1e-2:
 
38
  yield output
39
  return output
40
 
 
 
 
41
  mychatbot = gr.Chatbot(
42
  avatar_images=["./user.png", "./botm.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True,)
43