Erginous_project / summary.py
jobanpreet123's picture
Upload 3 files
11152bc verified
raw
history blame
663 Bytes
from langchain_groq import ChatGroq
from langchain.prompts import ChatPromptTemplate
from langchain_core.output_parsers import StrOutputParser
from dotenv import load_dotenv
load_dotenv()
model = ChatGroq(temperature=0, model_name="mixtral-8x7b-32768") #mixtral-8x7b-32768
def summarize(transcription):
template="""You are provided with transcription of a meeting between different people. Your task is to summarize this transcription
Transcription: {transcription}"""
prompt = ChatPromptTemplate.from_template(template)
chain = prompt | model | StrOutputParser()
summary=chain.invoke({"transcription":transcription})
return summary