Update app.py

app.py

@@ -1,7 +1,8 @@
+import os
 import gradio as gr
 from transformers import pipeline
 
-client = pipeline("text-generation", model="upstage/Llama-2-70b-instruct", api_url=API_URL)
+client = pipeline("text-generation", model="upstage/Llama-2-70b-instruct", api_url=os.environ['API_URL'])
 
 def format_chat_prompt(message, chat_history, instruction):
     prompt = f"System:{instruction}"