Update app.py

app.py

@@ -5,18 +5,22 @@ import keras_nlp
 import PyPDF2
 import docx2txt
 
+hf_username = huggingface_hub.whoami()['name']
+hf_url = f'hf://{hf_username}/bart_billsum'
+
 # Load your Keras model
 @st.cache_resource 
 def load_model_and_preprocessor():
-    bart_billsum = keras_nlp.models.BartSeq2SeqLM.from_preset('Grey01/bart_billsum')
+    bart_billsum = keras_nlp.models.BartSeq2SeqLM.from_preset(f'hf://{hf_username}/bart_billsum')
     
     # Load the default BART preprocessor (assuming you saved its configuration)
-    preprocessor = keras_nlp.models.BartSeq2SeqLMPreprocessor.from_preset('bart_base_en', encoder_sequence_length=512,
-    decoder_sequence_length=128,)
+    #preprocessor = keras_nlp.models.BartSeq2SeqLMPreprocessor.from_preset('bart_base_en', encoder_sequence_length=512,
+    
+    #decoder_sequence_length=128,)
     
-    return model, preprocessor
+    return model
 
-model, preprocessor = load_model_and_preprocessor()
+model = load_model_and_preprocessor()
 
 st.title("SummarizeIt")