Update app.py

app.py

@@ -21,7 +21,12 @@ def image_to_base64(image):
     image.save(buffered, format="JPEG")
     return base64.b64encode(buffered.getvalue()).decode()
 
-def handle_upload(image, prompt):
+def base64_to_image(base64_str):
+    image_data = base64.b64decode(base64_str)
+    return Image.open(BytesIO(image_data))
+
+def handle_upload(base64_image, prompt):
+    image = base64_to_image(base64_image)
     upscaled_image = upscale_image(image, prompt)
     base64_str = image_to_base64(upscaled_image)
     return base64_str
@@ -32,11 +37,11 @@ def main():
 
         with gr.Row():
             with gr.Column(scale=1):
-                image_input = gr.Image(type="pil", label="Low-Resolution Image")
+                image_input = gr.Textbox(label="Base64 Encoded Low-Resolution Image")
                 prompt_input = gr.Textbox(label="Prompt", value="a white cat")
 
                 upload_btn = gr.Button("Upload and Upscale")
-                base64_output = gr.Textbox(label="Base64 Encoded Image")
+                base64_output = gr.Textbox(label="Base64 Encoded Upscaled Image")
 
                 upload_btn.click(fn=handle_upload, inputs=[image_input, prompt_input], outputs=[base64_output])