Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import torch
-from detect import detect
+from detect import *
 from PIL import Image
 from io import *
 import glob
@@ -39,7 +39,7 @@ def imageInput(device, src):
                 f.write(image_file.getbuffer())
 
             #call Model prediction--
-            model = torch.hub.load('ultralytics/yolov5', 'custom', path='models/yoloTrained.pt', force_reload=True) 
+            model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True) 
             model.cuda() if device == 'cuda' else model.cpu()
             pred = model(imgpath)
             pred.render()  # render bbox in image
@@ -101,8 +101,7 @@ def main():
     st.sidebar.markdown("https://github.com/thepbordin/Obstacle-Detection-for-Blind-people-Deployment")
     if option == "Image":    
         imageInput(deviceoption, datasrc)
-    elif option == "Video": 
-        videoInput(deviceoption, datasrc)
+