Update app.py

app.py

@@ -42,7 +42,7 @@ class PromptRefiner:
             {"role": "user", "content": selected_meta_prompt.replace("[Insert initial prompt here]", prompt_input.text)}
         ]
         response = self.client.chat_completion(
-            model="meta-llama/Meta-Llama-3-70B-Instruct",
+            model=prompt_refiner_model,
             messages=messages,
             max_tokens=2000,
             temperature=0.8
@@ -136,8 +136,7 @@ class GradioInterface:
                         "NousResearch/Hermes-3-Llama-3.1-8B",
                         "HuggingFaceH4/zephyr-7b-alpha",
                         "meta-llama/Llama-2-7b-chat-hf",
-                        "microsoft/Phi-3.5-mini-instruct",
-                        "Qwen/Qwen2.5-1.5B-Instruct"
+                        "microsoft/Phi-3.5-mini-instruct"
                     ],
                     value="meta-llama/Meta-Llama-3-70B-Instruct",
                     label="Choose the Model to apply to the prompts (the one you will used)"
@@ -200,6 +199,7 @@ if __name__ == '__main__':
         raise ValueError("HF_API_TOKEN not found in environment variables")
 
     metadone = os.getenv('metadone')
+    prompt_refiner_model = os.getenv('prompt_refiner_model')
     echo_prompt_refiner = os.getenv('echo_prompt_refiner')
     metaprompt1 = os.getenv('metaprompt1')   
     loic_metaprompt = os.getenv('loic_metaprompt')