Update app.py

app.py

@@ -1,6 +1,4 @@
 import gradio as gr
-import torch
-from torchvision import transforms
 from PIL import Image
 import requests
 from diffusers import StableDiffusionPipeline
@@ -32,11 +30,7 @@ def merge_tags(tags1, tags2):
 # Gradio app functions
 def process_image(image, mode):
     # Convert the image to a format suitable for the models
-    transform = transforms.Compose([
-        transforms.Resize((256, 256)),
-        transforms.ToTensor()
-    ])
-    image = transform(image).unsqueeze(0)
+    image = image.resize((256, 256))
 
     if mode == "Anime":
         is_anime, similar_images, original_tags = check_anime_image(image)