Spaces:
Sleeping
Sleeping
| from langchain_community.llms import OpenAI | |
| import os | |
| import PIL | |
| import numpy as np | |
| from PIL import Image, ImageDraw | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| import streamlit as st | |
| import easyocr | |
| from langchain.prompts import PromptTemplate | |
| st.header("Summarizer") | |
| reader = easyocr.Reader(['en']) | |
| uploaded_file=st.file_uploader("Upload image:") | |
| llm=OpenAI(openai_api_key=os.getenv("OPENAI_API_KEY"),temperature=0.7) | |
| pt=PromptTemplate(input_variables=["text"],template="Read the following text: {text} and give the important points in the form of bullet points. ") | |
| from langchain.chains import LLMChain | |
| chain=LLMChain(llm=llm,prompt=pt) | |
| if uploaded_file is not None: | |
| image = Image.open(uploaded_file) | |
| st.image(image) | |
| result = reader.readtext(np.array(image), detail = 0) | |
| st.subheader("OCR result:") | |
| st.write(result) | |
| st.subheader("Summary: ") | |
| st.write(chain.invoke(result)["text"]) |