Update src/streamlit_app.py
Browse files- src/streamlit_app.py +3 -3
src/streamlit_app.py
CHANGED
|
@@ -53,7 +53,7 @@ def read_uploaded_file(uploaded_file):
|
|
| 53 |
def load_llm():
|
| 54 |
pipe = pipeline(
|
| 55 |
"text2text-generation",
|
| 56 |
-
model="google/flan-t5-
|
| 57 |
max_length=256,
|
| 58 |
temperature=0.7,
|
| 59 |
top_p=0.95
|
|
@@ -85,7 +85,7 @@ with st.sidebar:
|
|
| 85 |
|
| 86 |
# Add sample file download button
|
| 87 |
st.download_button(
|
| 88 |
-
label="
|
| 89 |
data=SAMPLE_TEXT,
|
| 90 |
file_name="sample_agri.txt",
|
| 91 |
mime="text/plain"
|
|
@@ -122,7 +122,7 @@ with st.sidebar:
|
|
| 122 |
|
| 123 |
st.session_state.document_processed = True
|
| 124 |
st.session_state.chat_history = []
|
| 125 |
-
st.success(f"
|
| 126 |
st.rerun()
|
| 127 |
else:
|
| 128 |
st.error("No content found in file.")
|
|
|
|
| 53 |
def load_llm():
|
| 54 |
pipe = pipeline(
|
| 55 |
"text2text-generation",
|
| 56 |
+
model="google/flan-t5-large",
|
| 57 |
max_length=256,
|
| 58 |
temperature=0.7,
|
| 59 |
top_p=0.95
|
|
|
|
| 85 |
|
| 86 |
# Add sample file download button
|
| 87 |
st.download_button(
|
| 88 |
+
label="📄 Download Sample File",
|
| 89 |
data=SAMPLE_TEXT,
|
| 90 |
file_name="sample_agri.txt",
|
| 91 |
mime="text/plain"
|
|
|
|
| 122 |
|
| 123 |
st.session_state.document_processed = True
|
| 124 |
st.session_state.chat_history = []
|
| 125 |
+
st.success(f"Processed {len(docs)} text chunks!")
|
| 126 |
st.rerun()
|
| 127 |
else:
|
| 128 |
st.error("No content found in file.")
|