Update app.py

app.py

@@ -9,13 +9,14 @@ def preprocess_city_name_and_generate_insight(city_name):
         return city_name, "Hugging Face token is not set in the environment variables."
     
     try:
-        client = InferenceClient(model_id="gpt2", api_token=token)
-        # Construct the prompt for GPT-2
+
+        client = InferenceClient(token=token)
+
         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."
+
+        result = client(inputs=prompt, model_id="gpt2")
+
+        insight = result['generated_text'].strip() if 'generated_text' in 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."
@@ -23,7 +24,6 @@ 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")