Spaces:
Sleeping
Sleeping
File size: 911 Bytes
d9e74df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
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"]) |