Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-import requests
 from PIL import Image
 
 st.image("src/RezbinLogo.jpg", width=150)
@@ -9,20 +8,6 @@ st.write("Upload an image of your waste before throwing it.")
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
 
 if uploaded_file is not None:
-    # Show preview
     img = Image.open(uploaded_file)
     st.image(img, caption="Uploaded Image", use_container_width=True)
-
-    # Send to AI API
-    files = {"file": uploaded_file.getvalue()}
-    try:
-        response = requests.post("https://your-api-url.com/predict", files=files)
-        if response.status_code == 200:
-            result = response.json()
-            label = result.get("label", "Unknown item")
-            reward = result.get("reward", 0)
-            st.markdown(f'<div class="orange-background">This is a {label}. You earn {reward:.2f}</div>', unsafe_allow_html=True)
-        else:
-            st.error(f"Error: {response.status_code} - {response.text}")
-    except Exception as e:
-        st.error(f"Failed to connect to model: {e}")
+    st.markdown('<div style="background-color: orange; padding: 10px; border-radius: 5px;">This is a can. You earn 0.50</div>', unsafe_allow_html=True)