Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
-import re
 from gradio_client import Client
+import re
+import os
 
 def escape_special_characters(result):
     # Convert the result to a string if it's not already
@@ -10,7 +11,10 @@ def escape_special_characters(result):
     return result_str
 
 def predict(question):
-    client = Client("https://teamtonic-bulbiplantdoctor.hf.space/--replicas/k5mwx/")
+    hf_token = os.getenv('HUGGINGFACE_TOKEN')
+    if not hf_token:
+        raise ValueError("Hugging Face token not found in environment variables")
+    client = Client("https://teamtonic-bulbiplantdoctor.hf.space/--replicas/k5mwx/", hf_token=hf_token)
     result = client.predict(question, api_name="/predict")
     escaped_result = escape_special_characters(result)
     return escaped_result