Update app.py

app.py

@@ -16,17 +16,17 @@ import os
 from PIL import Image
 
 
-base_model="stabilityai/stable-diffusion-xl-base-1.0"
+# base_model="stabilityai/stable-diffusion-xl-base-1.0"
 
+# # base = DiffusionPipeline.from_pretrained(base_model,
+# #         torch_dtype=torch.float16, use_safetensors=True
+# # ).to("cuda")
 # base = DiffusionPipeline.from_pretrained(base_model,
 #         torch_dtype=torch.float16, use_safetensors=True
-# ).to("cuda")
-base = DiffusionPipeline.from_pretrained(base_model,
-        torch_dtype=torch.float16, use_safetensors=True
-)
-# lora_model_path = "./pytorch_weights.safetensors"
+# )
+# # lora_model_path = "./pytorch_weights.safetensors"
 
-# base.load_lora_weights(lora_model_path)
+# # base.load_lora_weights(lora_model_path)
 
 def swap_face(source_file, target_file,doFaceEnhancer):
 
@@ -90,37 +90,37 @@ def swap_face(source_file, target_file,doFaceEnhancer):
 
 def generate_image(prompt, height=1024, width=1024):
     
-    image = base(
-        prompt=prompt,
-        negative_prompt="(deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime), text, cropped, out of frame, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck",
-        num_inference_steps=40,
-        guidance_scale=3,
-        height=int(height),
-        width=int(width),
-        num_images_per_prompt=1,
-    )
+    # image = base(
+    #     prompt=prompt,
+    #     negative_prompt="(deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime), text, cropped, out of frame, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck",
+    #     num_inference_steps=40,
+    #     guidance_scale=3,
+    #     height=int(height),
+    #     width=int(width),
+    #     num_images_per_prompt=1,
+    # )
     
 
-    for i in range(len(image.images)):
-        # Save the generated images
-        original_image_path = f"./output{i}.png"
-        image.images[i].save(original_image_path)
-
-        # The source image for face swapping
-        source_image_path = "./roop/IMG_0991.png"
+    # for i in range(len(image.images)):
+    #     # Save the generated images
+    #     original_image_path = f"./output{i}.png"
+    #     image.images[i].save(original_image_path)
 
-        # Run the face swap and optionally face enhancement
-        swapped_image_path = swap_face(source_image_path, original_image_path, False)
+    #     # The source image for face swapping
+    #     source_image_path = "./roop/IMG_0991.png"
 
-        # Load the swapped image and add to the output list
-        swapped_image = Image.open(swapped_image_path)
+    #     # Run the face swap and optionally face enhancement
+    #     swapped_image_path = swap_face(source_image_path, original_image_path, False)
 
-        return swapped_image
+    #     # Load the swapped image and add to the output list
+    #     swapped_image = Image.open(swapped_image_path)
 
-examples = [
-    ["xkx man as superman. Uhd, 8k", 1024, 1024],
-    ["photo of xkx man on a beach. Amazing scenery, waves, sun.  Uhd, 8k", 1024, 1536],
-]
+    #     return swapped_image
+    return None
+# examples = [
+#     ["xkx man as superman. Uhd, 8k", 1024, 1024],
+#     ["photo of xkx man on a beach. Amazing scenery, waves, sun.  Uhd, 8k", 1024, 1536],
+# ]
 # Set up the Gradio interface
 interface = gr.Interface(
     fn=generate_image,
@@ -130,7 +130,7 @@ inputs=[
         gr.Number(label="Width")
     ],
     outputs=gr.Image(label="Image"),
-    examples=examples
+    # examples=examples
             
 )