Update app.py

app.py

@@ -15,15 +15,14 @@ from agents.prompts import load_prompts
 
 from Gradio_UI import GradioUI
 
-
-my_id = os.getenv("QWEN_URI")
-
+# Get the URI for the endpoint 
+endpoint_uri = load_huggingface_model
 
 # Test the endpoint
 
-if not is_huggingface_endpoint(my_id):
+if not is_huggingface_endpoint(endpoint_uri):
      create_failed_gradio_ui()
-    # Stop execution if the endpoint is not valid
+
 else: 
     pass 
     
@@ -33,7 +32,7 @@ else:
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id= my_id,
+    model_id= endpoint_uri,
      custom_role_conversions=None,
  )