Update app.py

app.py

@@ -48,7 +48,20 @@ model.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
 
 
 def greet(image):
-    return "Hello " + "name" + "!!"
+    preprocess = transforms.Compose([
+            transforms.Resize((128, 128)),
+            transforms.ToTensor(),
+            transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
+    ])
+
+    classes = ['cat', 'dog']
+
+    x = preprocess(image).unsqueeze(0)
+    x = model(x)
+    output = torch.nn.functional.softmax(x, dim=1)
+    
+    
+    return classes[probabilities.argmax(dim=1).item()]
 
 demo = gr.Interface(fn=greet, inputs="image", outputs="text")
 demo.launch()