Surendradjh commited on
Commit
7dda05d
·
verified ·
1 Parent(s): eb1cfa3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import streamlit as st
2
  import assemblyai as aai
3
  from database import db
 
4
 
5
  st.title("Project Shazam - Audio File Uploader")
6
 
@@ -23,3 +24,7 @@ else:
23
  query = transcript.text
24
  docs_chroma = db.similarity_search_with_score(query, k=3)
25
  context_text = "\n\n".join([doc.page_content for doc,_score in docs_chroma]
 
 
 
 
 
1
  import streamlit as st
2
  import assemblyai as aai
3
  from database import db
4
+ from model import prompt_template,model
5
 
6
  st.title("Project Shazam - Audio File Uploader")
7
 
 
24
  query = transcript.text
25
  docs_chroma = db.similarity_search_with_score(query, k=3)
26
  context_text = "\n\n".join([doc.page_content for doc,_score in docs_chroma]
27
+ prompt = prompt_template.format(context=context_text, question=query)
28
+ response_text = model.invoke(prompt)
29
+
30
+ st.write(response_text.content)