Update app.py

app.py

@@ -1,20 +1,21 @@
 import gradio as gr
 import requests
 import os
-from huggingface_hub import InferenceApi  # Correct import for inference
+from huggingface_hub import InferenceClient
 
 def preprocess_city_name_and_generate_insight(city_name):
     token = os.getenv("HF_TOKEN")
     if not token:
         return city_name, "Hugging Face token is not set in the environment variables."
-    
-    # Using the InferenceApi class correctly
-    client = InferenceApi("gpt2", token)
+
+    # Correctly initializing InferenceClient with the token
+    client = InferenceClient("gpt2", token=token)
     prompt = f"Generate a short, interesting fact about {city_name} in less than 10 words:"
-    
+
     try:
-        result = client(predict=prompt)  # Using predict or the correct method
-        insight = result[0]['generated_text'].strip() if result else "No insight generated."
+        # Assuming the GPT-2 model expects a key named 'inputs' for its input
+        result = client(predict={"inputs": prompt})
+        insight = result['generated_text'].strip() if result else "No insight generated."
     except Exception as e:
         print(f"Error generating insight: {str(e)}")
         insight = "Could not generate insight due to an error."
@@ -22,6 +23,7 @@ def preprocess_city_name_and_generate_insight(city_name):
     return city_name, insight
 
 
+
 def get_weather(city_name):
     corrected_city_name, city_insight = preprocess_city_name_and_generate_insight(city_name)
     API_Key = os.getenv("OPENWEATHER_API_KEY")