changes app.py

app.py

@@ -1,33 +1,15 @@
 import gradio as gr
 import os
-import torch
 
-def update_value(val):
-  return f'Value is set to {val}'
 
-def yolov7_inference(
-    image: gr.Image = None,
-    conf_threshold: gr.Slider = 0.20,
-):
+def image_mod(image):
+    return image.rotate(45)
 
-    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-    path = 'y7-prdef.pt'
-    model = torch.hub.load("WongKinYiu/yolov7","custom",f"{path}")
-    model.conf = conf_threshold
-    results = model([image], size=640)
-    return results.render()[0]
-        
-demo = gr.Blocks()
-
-with demo:
-
-  gr.Interface(
-    yolov7_inference,
+demo = gr.Interface(
+    image_mod,
     gr.Image(type="pil"),
     "image",
-    title = "The detection of jar lid defects using Yolov7",
-    description = "contact: [email protected]", 
-
+    flagging_options=["blurry", "incorrect", "other"],
     examples=[
         os.path.join(os.path.dirname(__file__), "example1.JPG"),
         os.path.join(os.path.dirname(__file__), "example2.JPG"),
@@ -35,8 +17,4 @@ with demo:
     ],
 )
 
-  md = gr.Markdown("Confidence Threshold")
-  conf_threshold = gr.Slider(minimum=0, maximum=1, step=0.1, label='Value')
-  conf_threshold.change(fn=update_value, inputs=conf_threshold, outputs=md)
-
-demo.launch(debug=False)
+demo.launch()