Update main.py

main.py

@@ -45,13 +45,26 @@ async def classify_image(file: UploadFile):
     
     return JSONResponse(content={"predicted_class": predicted_class_name})
 
-    
-st.title("تصنيف الصور باستخدام Streamlit")
 
-uploaded_file = st.file_uploader("رفع صورة", type=["jpg", "jpeg", "png"])
+
+
+# Streamlit UI
+st.title("Binary Image Classification")
+st.write("Upload an image to classify if it's a flower or not.")
+
+uploaded_file = st.file_uploader("Choose an image...", type="jpg")
 
 if uploaded_file is not None:
+    # Display uploaded image
     image = Image.open(uploaded_file)
-    st.image(image, caption='الصورة التي تم رفعها.', use_column_width=True)
-    st.write("")
-    st.write("تم تصنيف الصورة كـ", classify_image(image))
+    st.image(image, caption="Uploaded Image.", use_column_width=True)
+
+    # Send the image to FastAPI for classification
+    url = "http://127.0.0.1:7860/classify/"  # Replace with your FastAPI server URL
+      # Replace with your FastAPI server URL
+    files = {"file": uploaded_file}
+    response = requests.post(url, files=files)
+
+    if response.status_code == 200:
+        result = response.json()
+        st.write("Predicted Class:", result["predicted_class"])