.

app.py

@@ -93,7 +93,7 @@ CVE-2015-10011 is a vulnerability about OpenDNS OpenResolve improper log output
             )
 
         with gr.Row():
-            max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
+            max_new_tokens = gr.Slider(minimum=1, maximum=2048, value=3, step=1, label="Max new tokens")
             temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
             top_p = gr.Slider(
                 minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
@@ -103,7 +103,7 @@ CVE-2015-10011 is a vulnerability about OpenDNS OpenResolve improper log output
         selected_models = gr.CheckboxGroup(
             choices=list(model_options.keys()),
             label="Select exactly two model to compare",
-            value=["Foundation-Sec-8B"],  # Default models
+            value=list(model_options.keys())[:2],  # Default models
         )
 
         # Dynamic Response Section
@@ -133,9 +133,10 @@ CVE-2015-10011 is a vulnerability about OpenDNS OpenResolve improper log output
                         generate_text_local, 
                         model_path, 
                         prompt, 
-                        max_tokens, 
-                        temperature, 
-                        top_p
+                        max_new_tokens = max_new_tokens,
+                        do_sample = True,
+                        temperature = temperature, 
+                        top_p = top_p
                     )
                     futures.append(future)
                     futures_to_model[future] = model_name
@@ -145,7 +146,11 @@ CVE-2015-10011 is a vulnerability about OpenDNS OpenResolve improper log output
                     model_name = futures_to_model[future]
                     responses[model_name] = future.result()
             
-            return responses.get(selected_models[0], ""), responses.get(selected_models[1], "")
+            # モデル名を冒頭に付加して返す
+            model1_output = f"{selected_models[0]} Output:\n\n{responses.get(selected_models[0], '')}"
+            model2_output = f"{selected_models[1]} Output:\n\n{responses.get(selected_models[1], '')}"
+            
+            return model1_output, model2_output
             
         # Add a button for generating responses
         submit_button = gr.Button("Generate Responses")