Update app.py

app.py

@@ -16,10 +16,10 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
 # Predict Function
 def predict(img):
   # Apply Transformations
-  # img=np.array(img)
+  img=np.array(img)
   img=transform_gen(img).unsqueeze(0)
-  # img=img['image'].unsqueeze(0)
-  # Predic
+  
+  # Predict
   gen.eval()
   with torch.inference_mode():
     y_gen=gen(img)
@@ -27,6 +27,7 @@ def predict(img):
   y_gen=to_img(y_gen)
   return y_gen
 
+
 # Gradio App
 title="Image Segmentation GAN"
 description="This segments a Normal Image"