Wakkamaruh commited on
Commit
0e81829
verified
1 Parent(s): 8d263d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -1,5 +1,4 @@
1
  from PIL import Image
2
- import requests
3
  import gradio as gr
4
  from transformers import BlipProcessor, BlipForConditionalGeneration
5
 
@@ -8,9 +7,9 @@ model_id = "Salesforce/blip-image-captioning-large"
8
  model = BlipForConditionalGeneration.from_pretrained(model_id)
9
  processor = BlipProcessor.from_pretrained(model_id)
10
 
11
- def launch(input):
12
- # Obtener la imagen desde la URL de entrada
13
- image = Image.open(requests.get(input, stream=True).raw).convert('RGB')
14
 
15
  # Procesar la imagen y generar el texto
16
  inputs = processor(image, return_tensors="pt")
@@ -25,6 +24,6 @@ def launch(input):
25
  # Devolver tanto la imagen como el texto
26
  return image, result
27
 
28
- # Cambiar la configuraci贸n de Gradio para que la salida incluya imagen y texto
29
- iface = gr.Interface(launch, inputs="text", outputs=["image", "text"])
30
  iface.launch()
 
1
  from PIL import Image
 
2
  import gradio as gr
3
  from transformers import BlipProcessor, BlipForConditionalGeneration
4
 
 
7
  model = BlipForConditionalGeneration.from_pretrained(model_id)
8
  processor = BlipProcessor.from_pretrained(model_id)
9
 
10
+ def launch(input_file):
11
+ # Abrir la imagen subida por el usuario
12
+ image = Image.open(input_file).convert('RGB')
13
 
14
  # Procesar la imagen y generar el texto
15
  inputs = processor(image, return_tensors="pt")
 
24
  # Devolver tanto la imagen como el texto
25
  return image, result
26
 
27
+ # Cambiar la configuraci贸n de Gradio para que la entrada sea un archivo y la salida incluya imagen y texto
28
+ iface = gr.Interface(launch, inputs="file", outputs=["image", "text"])
29
  iface.launch()