Update app.py

app.py

@@ -14,7 +14,7 @@ classes=["Other","Anger","Contempt","Happiness","Surprise"]
 # load a resnet18 model pretrained on ImageNet
 # and turn off autograd on model's parameters
 def load_model(idx):
-  model = torch.jit.load('/content/model_2_60acc.pt',map_location=torch.device('cpu')).eval()
+  model = torch.jit.load('model_2_60acc.pt',map_location=torch.device('cpu')).eval()
   for param in model.parameters():
     param.requires_grad = False
   return model
@@ -68,7 +68,7 @@ def main():
         with gr.Box():
             input_image = gr.Image(type="pil", label="Input Image")
             example_images = gr.Dataset(components=[input_image],
-                                        samples=[['/content/ogtest0.png'],['/content/ogtest1.png'],['/content/ogtest2.png'],['/content/ogtest3.png'],['/content/ogtest4.png'],['/content/ogtest5.png']]
+                                        samples=[['ogtest0.png'],['ogtest1.png'],['ogtest2.png'],['ogtest3.png'],['ogtest4.png'],['ogtest5.png']]
                                         ,type="index")
 
             with gr.Row():