Update app.py

app.py

@@ -10,7 +10,7 @@ import gradio as gr
 #model = torch.jit.load('https://huggingface.co/spaces/yuhe6/final_project/blob/main/Net_Rotate9.pth').eval().to(device)
 
 
-model = torch.jit.load('Net2_Blur_jit.pt', map_location = torch.device('cpu'))
+model = torch.jit.load('Net2_Flip_jit.pt', map_location = torch.device('cpu'))
 model.eval()
 
 model_categories = ["cat","dog"] # verify order