Update app.py

app.py

@@ -4,6 +4,8 @@ import gradio as gr
 
 api_key = os.environ.get("OPENAI_API_KEY")
 base_url = os.environ.get("OPENAI_API_BASE")
+model_name = os.environ.get("MODEL_NAME")
+
 client = OpenAI(api_key=api_key, base_url=base_url)
 
 def predict(message, history):
@@ -13,7 +15,7 @@ def predict(message, history):
         history_openai_format.append({"role": "assistant", "content":assistant})
     history_openai_format.append({"role": "user", "content": message})
   
-    response = client.chat.completions.create(model='Llama-3-8B-UltraMedical',
+    response = client.chat.completions.create(model=model_name,
     messages= history_openai_format,
     temperature=1.0,
     stop=["<|eot_id|>"],