Update app.py

app.py

@@ -33,7 +33,7 @@ if uploaded_image:
 
         # Preprocess the image
         img = image.resize((384, 288))
-        img = np.array(img) / 100.
+        img = np.array(img) / 255.
         img = np.expand_dims(np.transpose(img, (2, 0, 1)), axis=0)
         img = torch.from_numpy(img)
         img = img.type(torch.FloatTensor).to(device)