Update app.py

app.py

@@ -145,8 +145,8 @@ import random
 
 
 
-def load_models(inp):
-    return gr.update(label=models[inp])
+# def load_models(inp):
+#     return gr.update(label=models[inp])
 
 def format_prompt(message, history, cust_p):
     prompt = ""
@@ -157,7 +157,7 @@ def format_prompt(message, history, cust_p):
     prompt+=cust_p.replace("USER_INPUT",message)
     return prompt
 
-def chat_inf(system_prompt,prompt,history,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,cust_p):
+def chat_inf(system_prompt,prompt,history,memory,seed,temp,tokens,top_p,rep_p,chat_mem,cust_p):
     hist_len=0
     if not history:
         history = []
@@ -214,7 +214,7 @@ def check_rand(inp,val):
     
 with gr.Blocks() as app:
     memory=gr.State()
-    gr.HTML("""<center><h1 style='font-size:xx-large;'>Gemma Gemini Multimodal Chatbot</h1><br><h2>Gemini Sprint submission by Rishiraj Acharya. Uses Google's Gemini 1.0 Pro Vision multimodal model from Vertex AI with Google's Gemma 7B Instruct model from Hugging Face. Google Cloud credits are provided for this project.</h2>""")
+    gr.HTML("""<center><h1 style='font-size:xx-large;'>Gemma Gemini Multimodal Chatbot</h1><br><h3>Gemini Sprint submission by Rishiraj Acharya. Uses Google's Gemini 1.0 Pro Vision multimodal model from Vertex AI with Google's Gemma 7B Instruct model from Hugging Face. Google Cloud credits are provided for this project.</h3>""")
     chat_b = gr.Chatbot(show_label=True, show_share_button=True, show_copy_button=True, likeable=True, layout="bubble", bubble_full_width=False)
     with gr.Group():
         with gr.Row():
@@ -230,7 +230,7 @@ with gr.Blocks() as app:
                         with gr.Group():
                             stop_btn=gr.Button("Stop")
                             clear_btn=gr.Button("Clear")                
-                client_choice=gr.Dropdown(label="Models",type='index',choices=[c for c in models],value=models[0],interactive=True)
+                # client_choice=gr.Dropdown(label="Models",type='index',choices=[c for c in models],value=models[0],interactive=True)
             with gr.Column(scale=1):
                 with gr.Group():
                     rand = gr.Checkbox(label="Random Seed", value=True)
@@ -242,11 +242,11 @@ with gr.Blocks() as app:
                     chat_mem=gr.Number(label="Chat Memory", info="Number of previous chats to retain",value=4)
 
     
-    client_choice.change(load_models,client_choice,[chat_b])
-    app.load(load_models,client_choice,[chat_b])
+    # client_choice.change(load_models,client_choice,[chat_b])
+    # app.load(load_models,client_choice,[chat_b])
     
-    chat_sub=inp.submit(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt],[chat_b,memory])
-    go=btn.click(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt],[chat_b,memory])
+    chat_sub=inp.submit(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt],[chat_b,memory])
+    go=btn.click(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt],[chat_b,memory])
     
     stop_btn.click(None,None,None,cancels=[go,chat_sub])
     clear_btn.click(clear_fn,None,[inp,sys_inp,chat_b,memory])