Update app.py
Browse files
app.py
CHANGED
|
@@ -22,10 +22,14 @@ def generate_caption(image):
|
|
| 22 |
else:
|
| 23 |
return f"Error: {response.status_code} - {response.text}"
|
| 24 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
|
| 26 |
#Open the images
|
| 27 |
-
Image1=Image.open('dlaima/Multiple_Image_captioning/main/image1.jpg')
|
| 28 |
-
|
| 29 |
#Image3=Image.open('https://huggingface.co/spaces/dlaima/Multiple_Image_captioning/resolve/main/image3.jpeg')
|
| 30 |
|
| 31 |
|
|
@@ -37,7 +41,7 @@ demo = gr.Interface(
|
|
| 37 |
inputs=gr.Image(type="pil", label="Upload Image"),
|
| 38 |
outputs=gr.Textbox(label="Generated Caption"),
|
| 39 |
#examples=[Image1, Image2, Image3],
|
| 40 |
-
examples=[
|
| 41 |
title="Image Captioning App",
|
| 42 |
description=(
|
| 43 |
"Upload an image or use one of the predefined samples to generate a caption. "
|
|
|
|
| 22 |
else:
|
| 23 |
return f"Error: {response.status_code} - {response.text}"
|
| 24 |
|
| 25 |
+
def load_image(img_url):
|
| 26 |
+
image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
|
| 27 |
+
return image
|
| 28 |
+
|
| 29 |
|
| 30 |
#Open the images
|
| 31 |
+
#Image1=Image.open('dlaima/Multiple_Image_captioning/main/image1.jpg')
|
| 32 |
+
Image2=Image.open('https://huggingface.co/spaces/dlaima/Multiple_Image_captioning/resolve/main/image2.jpeg')
|
| 33 |
#Image3=Image.open('https://huggingface.co/spaces/dlaima/Multiple_Image_captioning/resolve/main/image3.jpeg')
|
| 34 |
|
| 35 |
|
|
|
|
| 41 |
inputs=gr.Image(type="pil", label="Upload Image"),
|
| 42 |
outputs=gr.Textbox(label="Generated Caption"),
|
| 43 |
#examples=[Image1, Image2, Image3],
|
| 44 |
+
examples=[Image2],
|
| 45 |
title="Image Captioning App",
|
| 46 |
description=(
|
| 47 |
"Upload an image or use one of the predefined samples to generate a caption. "
|