Update app.py

app.py

@@ -13,19 +13,19 @@ to_img=T.ToPILImage()
 example_list = [["examples/" + example] for example in os.listdir("examples")]
 # example_list=['1.jpg','2.jpg']
 
-# Predict Function
 def predict(img):
-  # Apply Transformations
-  img=np.array(img)
-  img=transform_gen(img).unsqueeze(0)
-  
-  # Predict
-  gen.eval()
-  with torch.inference_mode():
-    y_gen=gen(img)
-  y_gen=y_gen[0]
-  y_gen=to_img(y_gen)
-  return y_gen
+    # Apply Transformations
+    img = transform_gen(img).unsqueeze(0)
+
+    # Predict
+    gen.eval()
+    with torch.inference_mode():
+        y_gen = gen(img)
+    y_gen = y_gen[0]
+    y_gen = to_img(y_gen)
+
+    return y_gen
+
 
 
 # Gradio App