Update app.py

app.py

@@ -3,7 +3,6 @@ import pickle
 from pydantic import BaseModel
 import streamlit as st
 import requests
-from requests.exceptions import ConnectionError
 
 app = FastAPI()
 
@@ -63,23 +62,14 @@ if __name__ == "__main__":
 
     submit = st.button("Predict")
     if submit:
-        payload = {
-            "Sepal_Length": sepal_length,
-            "Sepal_Width": sepal_width,
-            "Petal_Length": petal_length,
-            "Petal_Width": petal_width
-        }
+        payload = {"Sepal_Length": sepal_length, "Sepal_Width": sepal_width, "Petal_Length": petal_length, "Petal_Width": petal_width}
         prediction = st.empty()
         with st.spinner("Predicting..."):
-            try:
-                response = requests.post("http://localhost:8000/predict", json=payload)
-                if response.status_code == 200:
-                    prediction_result = response.json()
-                    prediction.success(f"Prediction: {prediction_result['prediction']}")
-                else:
-                    prediction.error("Prediction failed.")
-            except ConnectionError as e:
-                prediction.error("Failed to connect to the server. Make sure the FastAPI server is running.")
-                print(str(e))
+            response = requests.post("http://localhost:8000/predict", json=payload)
+            if response.status_code == 200:
+                prediction_result = response.json()
+                prediction.success(f"Prediction: {prediction_result['prediction']}")
+            else:
+                prediction.error("Prediction failed.")
 
     uvicorn_proc.kill()