Update src/streamlit_app.py

src/streamlit_app.py

@@ -51,23 +51,24 @@ if st.button("Predict Sales"):
     with st.spinner("Fetching prediction from backend..."):
         try:
             response = requests.post(
-                "https://lokiiparihar-superkartbackendmodaldeploy-xgboost.hf.space/predict",
+                "https://lokiiparihar-SuperkartBackendModalDeploy-XGBoost.hf.space/predict",
                 json=input_data
             )
             if response.status_code == 200:
-                result = response.json()
-                prediction = result.get("prediction")
+                try:
+                    result = response.json()
+                    prediction = result.get("prediction", None)
+                except ValueError:
+                    # Handle raw text response
+                    prediction = response.text
 
-                if prediction is not None:
-                    try:
-                        prediction = float(prediction)
-                        st.success(f"✅ Predicted Sales: **{prediction:.2f} units**")
-                    except (ValueError, TypeError):
-                        st.error(f"❌ Prediction format error: {prediction}")
-                else:
-                    st.error("❌ Prediction not returned by backend.")
+                try:
+                    prediction = float(prediction)
+                    st.success(f"✅ Predicted Sales: **{prediction:.2f} units**")
+                except (ValueError, TypeError):
+                    st.error(f"❌ Could not convert prediction to number: {prediction}")
             else:
                 st.error(f"❌ API Error: Status code {response.status_code}")
-                st.text(f"Response: {response.text}")
+                st.text(response.text)
         except Exception as e:
             st.error(f"❌ Request failed: {e}")