Update app.py

app.py

@@ -1,18 +1,24 @@
+
 import numpy as np
 
 import gradio as gr
 import os
+os.system("pip3 install torch")
+os.system("pip3 install collections")
+os.system("pip3 install torchvision")
+#os.system("pip3 install argparse")
+
 from PIL import Image
 import torch
 from torchvision import transforms
 from model_video import build_model
 import numpy as np
 import collections
-import argparse
+#import argparse
 
 net = build_model('cpu').to('cpu')
     #net=torch.nn.DataParallel(net)
-model_path = '/content/models/image_best.pth'
+model_path = 'image_best.pth'
 print(model_path)
 weight=torch.load(model_path,map_location=torch.device('cpu'))
 #print(type(weight))