dlaima commited on
Commit
045fc21
·
verified ·
1 Parent(s): be4c74e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
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
- #Image2=Image.open('https://huggingface.co/spaces/dlaima/Multiple_Image_captioning/resolve/main/image2.jpeg')
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=[Image1],
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. "