Update app.py

app.py

@@ -195,14 +195,14 @@ class GradioInterface:
             result.dict()
         )
 
-    def apply_prompts(self,original_prompt: str, refined_prompt: str):
-        original_output = self.prompt_refiner.apply_prompt(original_prompt)
-        refined_output = self.prompt_refiner.apply_prompt(refined_prompt)
+    def apply_prompts(self, original_prompt: str, refined_prompt: str, model: str):
+        original_output = self.prompt_refiner.apply_prompt(original_prompt, model)
+        refined_output = self.prompt_refiner.apply_prompt(refined_prompt, model)
         return original_output, refined_output
 
-    def launch(self):
-        self.interface.launch()
-
+    def launch(self, share=False):
+        self.interface.launch(share=share)
+        
 # Main code to run the application
 if __name__ == '__main__':
     api_key = os.getenv('GROQ_API_KEY')