Create app.py

app.py

@@ -0,0 +1,38 @@
+import gradio as gr
+import os
+import torch
+
+
+#Repo.clone_from("https://github.com/ultralytics/yolov5", "./content")
+#subprocess.run(["git","clone", "https://github.com/ultralytics/yolov5"])
+'''gdd.download_file_from_google_drive(file_id='1xJcr1zuuoC6bbu6wGvmyw5dl0mUEi-Jn',
+                                    dest_path='content/train_data.zip',
+                                    unzip=True)
+gdd.download_file_from_google_drive(file_id='1UJ9Y2mw2KBRY0tO-TnBs5RALLya6C2dW',
+                                    dest_path='content/video.zip',
+                                    unzip=True) '''                               
+#subprocess.run(["mv","content/custom_data.yaml","./yolov5/data"])         
+
+def find(name, path):
+ for root, dirs, files in os.walk(path):
+  if name in files:
+   return os.path.join(root, name)
+
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='/content/best.pt', force_reload=True)    
+def detect(inp):
+ output=model(inp)
+ #outPath=
+  #out_path='./yolov5/runs/detect/exp/'+inp_vid.split('/')[2]
+ return output
+
+inp=gr.inputs.Image() 
+output=gr.outputs.Image() 
+
+
+io=gr.Interface(fn=detect, inputs=inp, outputs=output, title='Party Symbol Detection',examples=['content/4.jpg'])
+io.launch(debug=True,share=True)
+  
+
+
+                                             
+