Update app.py

app.py

@@ -54,6 +54,7 @@ from huggingface_hub import HfApi
 from huggingface_hub import login
 from huggingface_hub import revision_exists
 
+
 api = HfApi()
 repo_name = "paralym/pangea_logs"
 
@@ -213,7 +214,7 @@ def add_message(history, message):
 
 
 @spaces.GPU
-def bot(history):
+def bot(history, temperature, top_p, max_output_tokens):
     print("### turn start history",history)
     print("### turn start conv",our_chatbot.conversation)
     text = history[-1][0]
@@ -350,10 +351,10 @@ def bot(history):
         inputs=input_ids,
         streamer=streamer,
         images=image_tensor,
-        max_new_tokens=1024,
         do_sample=True,
-        temperature=0.2,
-        num_beams=1,
+        temperature=temperature,
+        top_p=top_p,
+        max_new_tokens=max_output_tokens,
         use_cache=False,
         stopping_criteria=[stopping_criteria],
     )
@@ -385,7 +386,7 @@ def bot(history):
     for upload_img in all_image_path:
         api.upload_file(
             path_or_fileobj=upload_img,
-            path_in_repo=upload_img.replace("./logs/"),
+            path_in_repo=upload_img.replace("./logs/", ""),
             repo_id=repo_name,
             repo_type="dataset",
             # revision=revision,
@@ -394,7 +395,7 @@ def bot(history):
     # upload json
     api.upload_file(
         path_or_fileobj=get_conv_log_filename(),
-        path_in_repo=get_conv_log_filename().replace("./logs/"),
+        path_in_repo=get_conv_log_filename().replace("./logs/", ""),
         repo_id=repo_name,
         repo_type="dataset")
         
@@ -416,6 +417,31 @@ with gr.Blocks(
     gr.HTML(html_header)
     
     with gr.Column():
+        with gr.Accordion("Parameters", open=False) as parameter_row:
+                temperature = gr.Slider(
+                    minimum=0.0,
+                    maximum=1.0,
+                    value=0.7,
+                    step=0.1,
+                    interactive=True,
+                    label="Temperature",
+                )
+                top_p = gr.Slider(
+                    minimum=0.0,
+                    maximum=1.0,
+                    value=1,
+                    step=0.1,
+                    interactive=True,
+                    label="Top P",
+                )
+                max_output_tokens = gr.Slider(
+                    minimum=0,
+                    maximum=8192,
+                    value=4096,
+                    step=256,
+                    interactive=True,
+                    label="Max output tokens",
+                )
         with gr.Row():
             chatbot = gr.Chatbot([], elem_id="Pangea", bubble_full_width=False, height=750)
 
@@ -608,7 +634,7 @@ with gr.Blocks(
     chat_msg = chat_input.submit(
         add_message, [chatbot, chat_input], [chatbot, chat_input]
     )
-    bot_msg = chat_msg.then(bot, chatbot, chatbot, api_name="bot_response")
+    bot_msg = chat_msg.then(bot, [chatbot,temperature, top_p, max_output_tokens], chatbot, api_name="bot_response")
     bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
 
     # chatbot.like(print_like_dislike, None, None)