Update api.py

api.py

@@ -1,15 +1,11 @@
-import requests
+# Step 9: Create a Gradio interface
+iface = gr.Interface(
+    fn=predict_rooms,
+    inputs="number",
+    outputs="number",
+    title="Number of Rooms Predictor",
+    description="Enter the population to predict the number of rooms needed."
+)
 
-# Define the API endpoint for your Hugging Face model
-API_URL = "https://jpb8uaoih9-496ff2e9c6d22116-7863-colab.googleusercontent.com/"
-
-# Define the function to make predictions using the Hugging Face Inference API
-def predict_rooms(population):
-    payload = {"inputs": population}
-    response = requests.post(API_URL, json=payload)
-    result = response.json()
-    return result["predictions"][0]
-
-# Make predictions using the predict_rooms function
-prediction = predict_rooms(1000)
-print(prediction)
+# Step 10: Launch the Gradio interface
+iface.launch()