Talha812 commited on
Commit
e05da15
Β·
verified Β·
1 Parent(s): 766a4a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -6,6 +6,9 @@ import faiss
6
  import fitz # PyMuPDF
7
  from langchain_text_splitters import RecursiveCharacterTextSplitter
8
 
 
 
 
9
  # Configuration
10
  MODEL_NAME = "ibm-granite/granite-3.1-1b-a400m-instruct"
11
  EMBED_MODEL = "sentence-transformers/all-mpnet-base-v2"
@@ -115,7 +118,6 @@ def generate_answer(query, context):
115
  return tokenizer.decode(outputs[0], skip_special_tokens=True).split("<|assistant|>")[-1].strip()
116
 
117
  # Streamlit UI
118
- st.set_page_config(page_title="πŸ“š Smart Book Analyst", layout="wide")
119
  st.title("πŸ“š AI-Powered Book Analysis System")
120
 
121
  uploaded_file = st.file_uploader("Upload book (PDF or TXT)", type=["pdf", "txt"])
 
6
  import fitz # PyMuPDF
7
  from langchain_text_splitters import RecursiveCharacterTextSplitter
8
 
9
+ # 1. Set page config FIRST
10
+ st.set_page_config(page_title="πŸ“š Smart Book Analyst", layout="wide")
11
+
12
  # Configuration
13
  MODEL_NAME = "ibm-granite/granite-3.1-1b-a400m-instruct"
14
  EMBED_MODEL = "sentence-transformers/all-mpnet-base-v2"
 
118
  return tokenizer.decode(outputs[0], skip_special_tokens=True).split("<|assistant|>")[-1].strip()
119
 
120
  # Streamlit UI
 
121
  st.title("πŸ“š AI-Powered Book Analysis System")
122
 
123
  uploaded_file = st.file_uploader("Upload book (PDF or TXT)", type=["pdf", "txt"])