Update app.py

app.py

@@ -6,15 +6,23 @@ from timeit import default_timer as timer
 
 model,transform=create_model()
 model=model.to("cpu")
-model.load_state_dict(torch.load("deneme_modeli.pth"))
+model.load_state_dict(
+    torch.load(
+        f="deneme_modeli.pth",
+        map_location=torch.device("cpu"),  # load to CPU
+    )
+)
+
 class_names = ["pizza","steak","sushi"]
 def predict(model,image):
     start=timer()
-    image=transform(image).unsqueeze(0).to("cpu")
-    output=model(image)
-
-    pred={class_names[i]:torch.softmax(output,dim=1)[0][i].item() for i in range(len(class_names))}
-    td=timer()-start
+    image=transform(image).unsqueeze(0)
+    model.eval()
+    with torch.inference_mode:
+        output=model(image)
+    
+        pred={class_names[i]:torch.softmax(output,dim=1)[0][i].item() for i in range(len(class_names))}
+        td=timer()-start
     return pred,td
 
 inputs = gr.Image(type="pil", label = "Resim")