web.py

import streamlit as st
from generation import *
from generation_att import *

st.markdown('## Генерирую текстовое описание изображений')
st.markdown('made by Kolesnikov Dmitry')

image = st.file_uploader("Загрузите изображение", type=['png', 'jpg','jpeg'])
if image is not None:
    st.image(image)
    img = Image.open(image)

    st.markdown('Результат генерации описания (модель LSTM + ResNet):')
    text = predict(img)
    st.markdown('#### __' + text + '__')
    st.markdown('---')

    st.markdown('Результат генерации описания (модель LSTM + MobileNet + Attention):')
    text_att = predict_att(img)
    print(text_att )
    st.markdown('#### __' + text_att + '__')