Update app.py

app.py

@@ -7,13 +7,9 @@ import io
 
 pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0")
 
-def generate_image(uploaded_file, surgery_option, additional_info):
-    if uploaded_file is not None:
-        if isinstance(uploaded_file, bytes):
-            pil_image = Image.open(io.BytesIO(uploaded_file)).convert("RGB")
-        else:
-            pil_image = uploaded_file.convert("RGB")
-        
+def generate_image(uploaded_image, surgery_option, additional_info):
+    if uploaded_image is not None:
+        pil_image = uploaded_image.convert("RGB")
         init_image = load_image(pil_image).convert("RGB")
         prompt = f"generate image of how this person would look after {surgery_option} also use this additional information {additional_info}"
         image = pipeline(prompt, image=init_image).images[0]
@@ -26,7 +22,7 @@ surgery_options = ["Facelift", "Rhinoplasty", "Chin Augmentation"]
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Image(type="auto", label="Upload an Image"),
+        gr.Image(type="pil", label="Upload an Image"),
         gr.Dropdown(surgery_options, label="Select Surgery Option"),
         gr.Textbox(label="Additional Information")
     ],
@@ -36,3 +32,4 @@ iface = gr.Interface(
 )
 
 iface.launch()
+