Silence1412 commited on
Commit
4999ece
1 Parent(s): 1ac37c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,14 +1,18 @@
1
  import streamlit as st
2
  import os
3
  from Chat_with_pdf_OpenAI import openai_pdf
 
4
 
5
  def main():
6
  st.header("Chat with your PDF")
7
  LLM_model = st.selectbox("Select Model",("OpenAI",
 
8
  "TODO: ADD MORE..."))
9
  create = st.button("Selected Model")
10
  if create and LLM_model == "OpenAI":
11
  openai_pdf()
 
 
12
 
13
  if __name__ == '__main__':
14
  main()
 
1
  import streamlit as st
2
  import os
3
  from Chat_with_pdf_OpenAI import openai_pdf
4
+ from Chat_with_pdf_LLM import LLM_pdf
5
 
6
  def main():
7
  st.header("Chat with your PDF")
8
  LLM_model = st.selectbox("Select Model",("OpenAI",
9
+ "google/flan-t5-large",
10
  "TODO: ADD MORE..."))
11
  create = st.button("Selected Model")
12
  if create and LLM_model == "OpenAI":
13
  openai_pdf()
14
+ if create and LLM_model != "OpenAI":
15
+ LLM_pdf(LLM_model)
16
 
17
  if __name__ == '__main__':
18
  main()