Update app.py

app.py

@@ -10,7 +10,7 @@ tinier = AutoModelForCausalLM.from_pretrained("afrizalha/Sasando-1-7M", token=hf
 
 desc = """Sasando-1 is a tiny, highly experimental text generator built using the Phi-3 architecture. It comes with two variations of microscopic sizes: 7M and 25M parameters. It is trained on a tightly-controlled Indo4B dataset filtered to only have 18000 unique words. The method is inspired by Microsoft's TinyStories paper which demonstrates that a tiny language model can produce fluent text when trained on tightly-controlled dataset.\n\nTry prompting with two simple words, and let the model continue. Fun examples provided below."""
 
-def generate(starting_text=None, choice=None, temp=None, top_p=None, info=False):
+def generate(starting_text=None, choice=None, temp=None, top_p=None):
     if info:
         return desc
     
@@ -18,6 +18,8 @@ def generate(starting_text=None, choice=None, temp=None, top_p=None, info=False)
         model = tinier
     elif choice == '25M':
         model = tiny
+    elif choice == "Info":
+        return desc
     
     results = []
     for i in range(5):
@@ -36,18 +38,17 @@ def generate(starting_text=None, choice=None, temp=None, top_p=None, info=False)
 
 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="Select model", value="7M")
-    info_button = gr.Button("Info")
+    choice = gr.Radio(["7M", "25M", "Info"], label="Select model", info="Built with the Phi-3 architecture", value='Info')
     
-    with gr.Row():
-        temp = gr.Slider(label="Temperature", minimum=0.05, maximum=1.0, step=0.05, value=0.7)
-        top_p = gr.Slider(label="Top P", minimum=0.05, maximum=1.0, step=0.05, value=0.5)
+    with gr.Row() as slider_options:
+        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, temp, top_p, info_button],
+        inputs=[starting_text, choice, slider_options],
         outputs=[res],
         allow_flagging="never",
         title="Sasando-1",
@@ -57,5 +58,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         ["gue"], ["presiden"], ["cinta adalah"], ["allah, aku"], ["dia marah karena"],
         ["inflasi"], ["kolam renang"], ["messi"], ["jalan-jalan"], ["komputer itu"]
     ], [starting_text])
-
 app.launch()