Update app.py

app.py

@@ -1,21 +1,21 @@
 import gradio as gr
 import requests
-import os
 from huggingface_hub import InferenceClient
+import os
 
 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."
-
-    # 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:
-        # 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."
+        client = InferenceClient(model_id="gpt2", api_token=token)
+        # Construct the prompt for GPT-2
+        prompt = f"Generate a short, interesting fact about {city_name} in less than 10 words."
+        # Make the prediction request
+        result = client.predict(inputs=prompt)
+        # Process the result
+        insight = result[0]['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."