IgorBogdan commited on
Commit
98c4ec7
·
verified ·
1 Parent(s): dfdde39

create app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -0
app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import BlipProcessor, BlipForConditionalGeneration
3
+ from PIL import Image
4
+
5
+ # Загрузка модели и процессора
6
+ processor = BlipProcessor.from_pretrained("mistralai/Pixtral-12B-2409")
7
+ model = BlipForConditionalGeneration.from_pretrained("mistralai/Pixtral-12B-2409")
8
+
9
+ # Заголовок приложения
10
+ st.title("Генерация описаний изображений")
11
+
12
+ # Загрузка изображения
13
+ uploaded_file = st.file_uploader("Выберите изображение...", type=["jpg", "jpeg", "png"])
14
+
15
+ if uploaded_file is not None:
16
+ # Открытие изображения
17
+ image = Image.open(uploaded_file)
18
+ st.image(image, caption='Загруженное изображение', use_column_width=True)
19
+
20
+ # Генерация описания
21
+ inputs = processor(images=image, return_tensors="pt")
22
+ out = model.generate(**inputs)
23
+ description = processor.decode(out[0], skip_special_tokens=True)
24
+
25
+ # Отображение результата
26
+ st.subheader("Описание:")
27
+ st.write(description)