Update app.py

app.py

@@ -9,7 +9,11 @@ pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-re
 
 def generate_image(uploaded_file, surgery_option, additional_info):
     if uploaded_file is not None:
-        pil_image = Image.open(io.BytesIO(uploaded_file))
+        if isinstance(uploaded_file, bytes):
+            pil_image = Image.open(io.BytesIO(uploaded_file)).convert("RGB")
+        else:
+            pil_image = uploaded_file.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]
@@ -17,17 +21,17 @@ def generate_image(uploaded_file, surgery_option, additional_info):
     else:
         return "No image uploaded."
 
-surgery_options = ["Face Lift", "Nose Correction", "Chin Implant"]
+surgery_options = ["Facelift", "Rhinoplasty", "Chin Augmentation"]
 
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Image(type="pil", label="Upload an Image"),
+        gr.Image(type="auto", label="Upload an Image"),
         gr.Dropdown(surgery_options, label="Select Surgery Option"),
         gr.Textbox(label="Additional Information")
     ],
     outputs=gr.Image(type="pil", label="Generated Image"),
-    title="SurgiLook.ai",
+    title="AI.HAIR - Version of Plastic Surgery by Bulut",
     description="Visualize the results of a surgical procedure before it happens."
 )