yetessam commited on
Commit
65b9821
·
verified ·
1 Parent(s): 29790d1

Adding print statements to log

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -13,6 +13,7 @@ import pytz # Had to give it permission in Code agent
13
  from tools.final_answer import FinalAnswerTool
14
  from tools.polite_guard import PoliteGuardTool
15
  from tools.web_search import DuckDuckGoSearchTool
 
16
 
17
  from Gradio_UI import GradioUI
18
 
@@ -51,6 +52,16 @@ my_id = 'Qwen/Qwen2.5-Coder-32B-Instruct'
51
 
52
  my_id = os.getenv("QWEN_URI")
53
 
 
 
 
 
 
 
 
 
 
 
54
  model = HfApiModel(
55
  max_tokens=2096,
56
  temperature=0.5,
 
13
  from tools.final_answer import FinalAnswerTool
14
  from tools.polite_guard import PoliteGuardTool
15
  from tools.web_search import DuckDuckGoSearchTool
16
+ from check_endpoint import is_huggingface_endpoint
17
 
18
  from Gradio_UI import GradioUI
19
 
 
52
 
53
  my_id = os.getenv("QWEN_URI")
54
 
55
+
56
+ # Test the endpoint
57
+
58
+ if is_huggingface_endpoint(my_id):
59
+ print("This is a Hugging Face Inference Endpoint.")
60
+ else:
61
+ print("This is NOT a Hugging Face Inference Endpoint.")
62
+
63
+
64
+
65
  model = HfApiModel(
66
  max_tokens=2096,
67
  temperature=0.5,