Update app.py (#2)

- Update app.py (7f1109c2255da6ef4d30fa4c0a4364b8cb70f2c7)


Co-authored-by: Tadesse Abateneh <[email protected]>

app.py

@@ -53,6 +53,10 @@ adapter.to(device)
 clip_model, _ = clip.load("ViT-B/32", device=device)
 
 def manipulate(input_image, caption):
+    if not isinstance(input_image, Image.Image):
+        input_image = Image.open(input_image).convert('RGB')
+    else:
+        input_image = input_image.convert('RGB')
     aligned_image = run_alignment(input_image)
     input_image = input_transforms(aligned_image)
     input_image = input_image.unsqueeze(0)