DHEIVER commited on
Commit
dbd941d
·
verified ·
1 Parent(s): d86e7e3

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -0
app.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import BlipProcessor, BlipForConditionalGeneration
3
+ from PIL import Image
4
+ import requests
5
+
6
+ # Carregar o modelo e o processador BLIP
7
+ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
8
+ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
9
+
10
+ def interpret_image(image):
11
+ # Converter a imagem para o formato PIL se necessário
12
+ if isinstance(image, str):
13
+ image = Image.open(requests.get(image, stream=True).raw)
14
+
15
+ # Processar a imagem e gerar a descrição
16
+ inputs = processor(image, return_tensors="pt")
17
+ out = model.generate(**inputs)
18
+
19
+ # Decodificar a saída para texto
20
+ description = processor.decode(out[0], skip_special_tokens=True)
21
+
22
+ return description
23
+
24
+ # Interface Gradio
25
+ iface = gr.Interface(
26
+ fn=interpret_image,
27
+ inputs=gr.Image(type="pil", label="Upload uma imagem"),
28
+ outputs=gr.Textbox(label="Descrição da Imagem"),
29
+ title="Interpretação de Imagens com BLIP",
30
+ description="Carregue uma imagem e o modelo BLIP irá gerar uma descrição."
31
+ )
32
+
33
+ # Iniciar o aplicativo
34
+ iface.launch()