CR7CAD commited on
Commit
d21b321
·
verified ·
1 Parent(s): 1167ad9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -47,15 +47,15 @@ def load_models():
47
  # Use pipeline if available, now using the updated model
48
  models['summarizer'] = pipeline(
49
  "summarization",
50
- model="marianna13/flan-t5-base-summarization",
51
  max_length=100,
52
  truncation=True
53
  )
54
  else:
55
  # Fall back to basic model loading using the updated summarization model
56
  try:
57
- models['summarizer_model'] = AutoModelForSeq2SeqLM.from_pretrained("marianna13/flan-t5-base-summarization")
58
- models['summarizer_tokenizer'] = AutoTokenizer.from_pretrained("marianna13/flan-t5-base-summarization")
59
  except Exception as e:
60
  st.error(f"Error loading summarization model: {e}")
61
  models['summarizer_model'] = None
 
47
  # Use pipeline if available, now using the updated model
48
  models['summarizer'] = pipeline(
49
  "summarization",
50
+ model="Falconsai/text_summarization",
51
  max_length=100,
52
  truncation=True
53
  )
54
  else:
55
  # Fall back to basic model loading using the updated summarization model
56
  try:
57
+ models['summarizer_model'] = AutoModelForSeq2SeqLM.from_pretrained("Falconsai/text_summarization")
58
+ models['summarizer_tokenizer'] = AutoTokenizer.from_pretrained("Falconsai/text_summarization")
59
  except Exception as e:
60
  st.error(f"Error loading summarization model: {e}")
61
  models['summarizer_model'] = None