Update app.py

app.py

@@ -22,8 +22,8 @@ gemma_pipe = pipeline(
 )
 
 # Initialize the Bloom model
-bloom_model = BloomForCausalLM.from_pretrained("bigscience/bloom-1b3")
-bloom_tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-1b3")
+bloom_model = BloomForCausalLM.from_pretrained("bigscience/bloom-560m")
+bloom_tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-560m")
 
 def extract_text_from_pdf(pdf_file):
     document = fitz.open(pdf_file)
@@ -122,4 +122,3 @@ iface = gr.Interface(
 )
 
 iface.launch()
-