Update app.py

app.py

@@ -93,7 +93,7 @@ class PromptRefiner:
             ]
 
             response = self.client.chat.completions.create(
-                model="llama-3.2-90b-text-preview",
+                model=model,
                 messages=messages,
                 max_tokens=8196, # Increased token limit
                 temperature=0.5
@@ -138,7 +138,19 @@ class GradioInterface:
                 outputs=[analysis_evaluation, refined_prompt, explanation_of_refinements, full_response_json]
             )
             with gr.Row():
-            #   apply_model=gr.Dropdown(["gpt-4o",'gpt-4-turbo'], value="gpt-4o", label="Model"),
+                apply_model = gr.Dropdown(
+                    [
+                        "llama-3.1-70b-versatile",
+                        "llama3-groq-70b-8192-tool-use-preview",
+                        "llama-3.2-90b-text-preview",
+                        "llama-3.2-90b-vision-preview",
+                        "llama3-groq-70b-8192-tool-use-preview",
+                        "mixtral-8x7b-32768"
+                    ],
+                    value="llama-3.1-70b-versatile",
+                    label="Choose Model"                
+                )
+                #   apply_model=gr.Dropdown(["gpt-4o",'gpt-4-turbo'], value="gpt-4o", label="Model"),
                 apply_button = gr.Button("Apply Prompts")
 
             with gr.Row():
@@ -151,7 +163,7 @@ class GradioInterface:
 
             apply_button.click(
                 fn=self.apply_prompts,
-                inputs=[prompt_text, refined_prompt],
+                inputs=[prompt_text, refined_prompt, apply_model],
                 outputs=[original_output, refined_output]
             )