Updated app.py

app.py

@@ -1,7 +1,43 @@
 import gradio as gr
+import requests
+from PIL import Image
+from io import BytesIO
 
-def greet(name):
-    return "Hello " + name + "!!"
+url = 'http://146.190.200.87:8000/predict'
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+def inference(input_image):
+    # Convert the PIL image to bytes
+    img_bytes = BytesIO()
+    input_image.save(img_bytes, format='JPEG')
+    
+    # Send the POST request with the image data as 'image' key
+    files = {'image': ('input.jpg', img_bytes.getvalue(), 'image/jpeg')}
+    response = requests.post(url, files=files)
+
+    # Check if the request was successful (status code 200)
+    if response.status_code == 200:
+        # Assuming the response is in JSON format, extract the category and probability information
+        result = response.json()
+        # Assuming the response contains category: probability pairs
+        # Sort the results by probability in descending order and take the top 5
+        sorted_results = sorted(result.items(), key=lambda x: x[1], reverse=True)[:5]
+        return sorted_results
+    else:
+        print("POST request failed with status code:", response.status_code)
+        return None
+
+title = "SeeFood102"
+description = "Gradio frontend for SeeFood102, the expansion edition of SeeFood101."
+
+examples = [
+    [Image.open('Screenshot 2023-05-05 085533.png')]
+]
+
+iface = gr.Interface(fn=inference,
+                     inputs=gr.Image(type="pil"),
+                     outputs=gr.Label(num_top_classes=5),
+                     title=title,
+                     description=description,
+                     examples=examples,
+                     analytics_enabled=False)
 iface.launch()