Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,29 +43,22 @@ def crop_face(base64_image):
|
|
| 43 |
except Exception as e:
|
| 44 |
return f"An error occurred: {str(e)}"
|
| 45 |
|
| 46 |
-
@spaces.GPU
|
| 47 |
def image_to_base64(image):
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
except Exception as e:
|
| 55 |
-
print("Error:", str(e))
|
| 56 |
-
return f"An error occurred: {str(e)}"
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
# Define the Gradio interfaces
|
| 61 |
|
| 62 |
|
|
|
|
| 63 |
base64_converter_interface = gr.Interface(
|
| 64 |
-
fn=image_to_base64,
|
| 65 |
-
inputs=gr.Image(),
|
| 66 |
-
outputs=
|
| 67 |
-
title="Image to Base64
|
| 68 |
-
description="Upload an image
|
| 69 |
)
|
| 70 |
|
| 71 |
face_crop_interface = gr.Interface(
|
|
@@ -77,4 +70,4 @@ face_crop_interface = gr.Interface(
|
|
| 77 |
)
|
| 78 |
|
| 79 |
if __name__ == "__main__":
|
| 80 |
-
gr.TabbedInterface([
|
|
|
|
| 43 |
except Exception as e:
|
| 44 |
return f"An error occurred: {str(e)}"
|
| 45 |
|
|
|
|
| 46 |
def image_to_base64(image):
|
| 47 |
+
# Convert PIL Image to bytes
|
| 48 |
+
buffered = io.BytesIO()
|
| 49 |
+
image.save(buffered, format="JPEG")
|
| 50 |
+
# Encode bytes to Base64 string
|
| 51 |
+
img_str = base64.b64encode(buffered.getvalue()).decode()
|
| 52 |
+
return img_str
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
|
| 54 |
|
| 55 |
+
# Define the Gradio interface using the updated syntax
|
| 56 |
base64_converter_interface = gr.Interface(
|
| 57 |
+
fn=image_to_base64,
|
| 58 |
+
inputs=gr.Image(type="pil"),
|
| 59 |
+
outputs=gr.Textbox(),
|
| 60 |
+
title="Image to Base64 Encoder",
|
| 61 |
+
description="Upload an image and convert it to a Base64 encoded string."
|
| 62 |
)
|
| 63 |
|
| 64 |
face_crop_interface = gr.Interface(
|
|
|
|
| 70 |
)
|
| 71 |
|
| 72 |
if __name__ == "__main__":
|
| 73 |
+
gr.TabbedInterface([base64_converter_interface, face_crop_interface], ["Convert to Base64","Crop Face"]).launch()
|