Update app.py

app.py

@@ -100,11 +100,11 @@ print("Loading model... This may take a few minutes.")
 generator = SketchToRealisticFace()
 print("Model loaded successfully!")
 
-def pil_to_base64(image):
-    buffer = BytesIO()
-    image.save(buffer, format="PNG")
-    img_str = base64.b64encode(buffer.getvalue()).decode()
-    return f"data:image/png;base64,{img_str}"
+# def pil_to_base64(image):
+#     buffer = BytesIO()
+#     image.save(buffer, format="PNG")
+#     img_str = base64.b64encode(buffer.getvalue()).decode()
+#     return f"data:image/png;base64,{img_str}"
 
 @spaces.GPU
 def generate_face(sketch_image, custom_prompt, seed):
@@ -116,12 +116,12 @@ def generate_face(sketch_image, custom_prompt, seed):
         # Generate the realistic face
         realistic_face, control_image = generator.generate(sketch_image=sketch_image,custom_prompt=custom_prompt,seed=seed_int)
         
-        #return realistic_face, control_image
+        return realistic_face, control_image
 
         # Convert to base64 (uncomment if you want base64 strings)
-        realistic_b64 = pil_to_base64(realistic_face)
-        control_b64 = pil_to_base64(control_image)
-        return realistic_b64, control_b64
+        # realistic_b64 = pil_to_base64(realistic_face)
+        # control_b64 = pil_to_base64(control_image)
+        # return realistic_b64, control_b64
         
     except Exception as e:
         print(f"Error: {str(e)}")