lamini models

app.py

@@ -7,12 +7,14 @@ template_by_step = """Question: {question}
 
 Answer: Let's think step by step."""
 
+models = ["MBZUAI/LaMini-Flan-T5-248M", "MBZUAI/LaMini-Flan-T5-783M"]
+
 
 def run(
     question: gr.Textbox = None,
-    repo_id: gr.Dropdown = "google/flan-t5-xxl",
+    repo_id: gr.Dropdown = models[0],
     temperature: gr.Slider = 0.5,
-    max_length: gr.Slider = 64,
+    max_length: gr.Slider = 512,
     by_steq: gr.Checkbox = False,
 ):
     template = template_by_step if by_steq else "{question}"
@@ -27,11 +29,10 @@ def run(
     return result
 
 
-models = ["MBZUAI/LaMini-Flan-T5-248M", "MBZUAI/LaMini-Flan-T5-783M"]
 inputs = [
     gr.Textbox(label="Question", lines=3),
     gr.Dropdown(choices=models,
-                value=models[0] if models else None, label="Model", allow_custom_value=True),
+                value=models[0], label="Model", allow_custom_value=True),
     gr.Slider(0.0, 1.0, value=0.5, step=0.05, label="Temperature"),
     gr.Slider(64, 1024, value=512, label="Max Length"),
     gr.Checkbox(label="Think step by step", value=False),