Update app.py

app.py

@@ -19,7 +19,7 @@ def generate(starting_text, choice, num_runs,temp,top_p):
         return desc
         
     results = []
-    for i in range(num_runs):
+    for i in range(5):
         inputs = tokenizer([starting_text], return_tensors="pt").to(model.device)
         outputs = model.generate(
             inputs=inputs.input_ids,
@@ -35,14 +35,14 @@ def generate(starting_text, choice, num_runs,temp,top_p):
 
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     starting_text = gr.Textbox(label="Starting text", value="cinta adalah")
-    choice = gr.Radio(["7M", "25M"], label="Model size", info="Built with the Phi-3 architecture")
-    num_runs = gr.Slider(label="Number of examples", minimum=1, maximum=10, step=1, value=5)
+    choice = gr.Radio(["7M", "25M", "Info"], label="Select model", info="Built with the Phi-3 architecture")
+    # num_runs = gr.Slider(label="Number of examples", minimum=1, maximum=10, step=1, value=5)
     temp = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, step=0.1, value=0.7)
     top_p = gr.Slider(label="Top P", minimum=0.1, maximum=1.0, step=0.1, value=0.5)
     res = gr.Textbox(label="Continuation")
     gr.Interface(
         fn=generate,
-        inputs=[starting_text,choice,num_runs,temp,top_p],
+        inputs=[starting_text,choice,temp,top_p],
         outputs=[res],
         allow_flagging="never",
         title="Sasando-1",