Delete feifeilib/feifeiimgtoimg.py

feifeilib/feifeiimgtoimg.py

@@ -1,35 +0,0 @@
-import gradio as gr
-from PIL import Image
-import numpy as np
-import random
-import spaces
-import torch.nn.functional as F
-import torch
-from diffusers import FluxImg2ImgPipeline
-from diffusers.utils import load_image
-
-dtype = torch.bfloat16
-device = "cuda" if torch.cuda.is_available() else "cpu"
-img_pipe = FluxImg2ImgPipeline.from_pretrained("aifeifei798/shuttle-3-diffusion",
-                                           torch_dtype=dtype)
-img_pipe = img_pipe.to(device)
-MAX_SEED = np.iinfo(np.int32).max
-
-@spaces.GPU()
-def feifeiimgtoimg(img_in_result,
-                   prompt,
-                   progress=gr.Progress(track_tqdm=True)):
-    seed = random.randint(0, MAX_SEED)
-    generator = torch.Generator().manual_seed(seed)
-    init_image = load_image(img_in_result)
-    images = img_pipe(prompt="",
-                      prompt_2=prompt,
-                      image=init_image,
-                      width=896,
-                      height=1152,
-                      num_inference_steps=4,
-                      strength=0.95,
-                      generator=generator,
-                      utput_type="pil",
-                      guidance_scale=3.5).images[0]
-    return images