Spaces:

islasher
/

Practica3_opc

Sleeping

islasher commited on Apr 22, 2024

Commit

1b51526

verified ·

1 Parent(s): 6b97e90

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ def predict(img):
     image = transforms.Resize((480,640))(img)
     tensor = transform_image(image=image)
     with torch.no_grad():
-        outputs = model(tensor)
     outputs = torch.argmax(outputs,1)

     image = transforms.Resize((480,640))(img)
     tensor = transform_image(image=image)
     with torch.no_grad():
+        outputs = learn.model(tensor)
     outputs = torch.argmax(outputs,1)