Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 
 def getResult(filename):
   model_name='best.pt'
-  model = torch.hub.load( "hubconf.py",'custom', source='local', path = model_name, force_reload = True)
+  model = torch.hub.load( 'custom', source='local', path = model_name, force_reload = True)
   img = "/content/"+filename+"" 
   results = model(img)
   return results
@@ -14,4 +14,4 @@ if uploaded_file is not None:
   image = uploaded_file.read()
   filename = uploaded_file.name
   resulted_img = getResult(filename)
-  img = st.image(resulted_img, caption='Sunrise by the mountains', use_column_width=True)
+  img = st.image(image, caption='Sunrise by the mountains', use_column_width=True)