g19_project / pages /Summarizer.py
Emil25's picture
Rename pages/Summary.py to pages/Summarizer.py
5cde23d
raw
history blame
1.39 kB
import streamlit as st
import requests
import os
# API для составления коспекта текста с использованием модели "bart-large-cnn"
API_URL_sum = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
headers = {"Authorization": os.getenv("api_token")}
# Функция для составления конспекта
def make_summary(payload):
response = requests.post(API_URL_sum, headers=headers, json=payload)
return response.json()
st.markdown('# :female-student: Персональный помощник для студентов')
st.divider()
st.markdown('# :blue_book: Конспект на английском языке')
st.markdown('## Введите текст на английском')
full_text = st.text_area(':book:', height=300, value='Введите свой текст на английском языке')
summary_button = st.button('Составить конспект')
if summary_button:
# Отправляем запросы через API для получения перевода слов и генерирования предложений
with st.spinner('...'):
summary_text = make_summary({"inputs": full_text})
with st.expander("Конспект"):
st.write(summary_text[0]['summary_text'])
st.success('Готово')
st.divider()