harish199 commited on
Commit
42162db
1 Parent(s): c1b9159

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ def llm_pipeline(filepath):
30
  'summarization',
31
  model = base_model,
32
  tokenizer = tokenizer,
33
- max_length = 500,
34
  min_length = 50)
35
  input_text = file_preprocessing(filepath)
36
  result = pipe_sum(input_text)
@@ -61,7 +61,7 @@ def main():
61
  if uploaded_file is not None:
62
  if st.button("Summarize"):
63
  col1, col2 = st.columns(2)
64
- filepath = "uploaded_file.name"
65
  with open(filepath, "wb") as temp_file:
66
  temp_file.write(uploaded_file.read())
67
  with col1:
 
30
  'summarization',
31
  model = base_model,
32
  tokenizer = tokenizer,
33
+ max_length = 5000,
34
  min_length = 50)
35
  input_text = file_preprocessing(filepath)
36
  result = pipe_sum(input_text)
 
61
  if uploaded_file is not None:
62
  if st.button("Summarize"):
63
  col1, col2 = st.columns(2)
64
+ filepath = "data/"+uploaded_file.name
65
  with open(filepath, "wb") as temp_file:
66
  temp_file.write(uploaded_file.read())
67
  with col1: