Update app.py

app.py

@@ -1,16 +1,23 @@
-import gradio as gr
+import streamlit as st
+import requests
 
-def predict(weight, mpg):
-    return f"Weight: {weight}, MPG: {mpg}"
+st.title("SuperKart Frontend")
 
-demo = gr.Interface(
-    fn=predict,
-    inputs=[
-        gr.Number(label="Enter weight"),
-        gr.Number(label="Enter mpg")
-    ],
-    outputs="text",
-    title="SuperKart Frontend"
-)
+weight = st.number_input("Enter weight", value=0.0)
+mpg = st.number_input("Enter mpg", value=0.0)
 
-demo.launch()
+if st.button("Predict"):
+
+    url = "https://muleex12-backkends.hf.space/run/predict"
+
+    payload = {
+        "data": [weight, mpg]
+    }
+
+    response = requests.post(url, json=payload)
+
+    if response.status_code == 200:
+        result = response.json()["data"][0]
+        st.success(f"Prediction: {result}")
+    else:
+        st.error("Error connecting to backend")