nicole-ait commited on
Commit
5639a50
1 Parent(s): 017da00

lamini models

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -27,12 +27,13 @@ def run(
27
  return result
28
 
29
 
 
30
  inputs = [
31
- gr.Textbox(label="Question"),
32
- gr.Dropdown(["google/flan-t5-xxl", "google/flan-t5-base"],
33
- value="google/flan-t5-xxl", label="Model", allow_custom_value=True),
34
  gr.Slider(0.0, 1.0, value=0.5, step=0.05, label="Temperature"),
35
- gr.Slider(20, 1000, value=64, label="Max Length"),
36
  gr.Checkbox(label="Think step by step", value=False),
37
  ]
38
 
@@ -43,6 +44,7 @@ examples = [
43
  ["What NFL team won the Super Bowl in the year Justin Bieber was born?"],
44
  ["Translate the following to French: There are so many plans"],
45
  ["Write an article to introduce machine learning"],
 
46
  ]
47
 
48
  title = "Langchain w/ HF Models"
 
27
  return result
28
 
29
 
30
+ models = ["MBZUAI/LaMini-Flan-T5-248M", "MBZUAI/LaMini-Flan-T5-783M"]
31
  inputs = [
32
+ gr.Textbox(label="Question", lines=3),
33
+ gr.Dropdown(choices=models,
34
+ value=models[0] if models else None, label="Model", allow_custom_value=True),
35
  gr.Slider(0.0, 1.0, value=0.5, step=0.05, label="Temperature"),
36
+ gr.Slider(64, 1024, value=512, label="Max Length"),
37
  gr.Checkbox(label="Think step by step", value=False),
38
  ]
39
 
 
44
  ["What NFL team won the Super Bowl in the year Justin Bieber was born?"],
45
  ["Translate the following to French: There are so many plans"],
46
  ["Write an article to introduce machine learning"],
47
+ ["Please let me know if you think the given place deserves to be visited and why: \"Beijing, China\""],
48
  ]
49
 
50
  title = "Langchain w/ HF Models"