karimD2 commited on
Commit
f32f32f
·
1 Parent(s): b9de7ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -26,7 +26,7 @@ pipe = pipeline(
26
  )
27
 
28
  generation_args = {
29
- "max_new_tokens": 500,
30
  "return_full_text": False,
31
  "temperature": 0.2,
32
  "do_sample": True,
@@ -44,14 +44,14 @@ examples = [
44
  'How are You?',
45
  "talk about your self",
46
  ]
47
- # gr.ChatInterface(
48
- # fn=phi3_fun,
49
- # title =title,
50
- # examples = examples
51
- # ).launch(debug=True)
52
- demo = gr.Interface(fn=phi3_fun, inputs="text", outputs="text",title =title,
53
- examples = examples)
54
- demo.launch()
55
 
56
  # output = pipe(messages, **generation_args)
57
  # print(output[0]['generated_text'])
 
26
  )
27
 
28
  generation_args = {
29
+ "max_new_tokens": 256,
30
  "return_full_text": False,
31
  "temperature": 0.2,
32
  "do_sample": True,
 
44
  'How are You?',
45
  "talk about your self",
46
  ]
47
+ gr.ChatInterface(
48
+ fn=phi3_fun,
49
+ title =title,
50
+ examples = examples
51
+ ).launch(debug=True)
52
+ # demo = gr.Interface(fn=phi3_fun, inputs="text", outputs="text",title =title,
53
+ # examples = examples)
54
+ # demo.launch()
55
 
56
  # output = pipe(messages, **generation_args)
57
  # print(output[0]['generated_text'])