Upload 3 files

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+from ultralytics import YOLO
+import torch
+# model = YOLO('best.pt')
+path = 'best.pt'
+model = torch.hub.load("WongKinYiu/yolov7","custom",path,trust_repo=True)
+
+def predict(input_image):
+    """
+    Predict model output
+    """
+    price = "0"
+    return [input_image, price]
+
+with gr.Blocks() as demo:
+    # Title
+    gr.Markdown(
+    """
+    <h1 align="center">AI Cafeteria Price Evaluator</h1>
+    """)
+
+    # Model Evaluation
+    gr.Interface(
+        fn=predict,
+        inputs=gr.Image(type="pil"),
+        outputs=[gr.Image(type="pil", label="Image Prediction"),
+                 gr.Textbox(type="text", label="Price Prediction")]
+    )
+
+if __name__ == "__main__":
+    demo.launch()

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e649c278fc179eb0b75e136f2ea86a58a2dd84a7f77dba4fe2b93987d56e63df
+size 75247994

requirements.txt

@@ -0,0 +1,2 @@
+torch
+ultralytics