Fixed model loading for Bloom

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 from PyPDF2 import PdfReader
 
 # Models and tokenizers setup
 models = {
     "Text Generator (Bloom)": {
-        "model": AutoModelForSeq2SeqLM.from_pretrained("bigscience/bloom-560m"),
+        "model": AutoModelForCausalLM.from_pretrained("bigscience/bloom-560m"),
         "tokenizer": AutoTokenizer.from_pretrained("bigscience/bloom-560m"),
     },
     "PDF Summarizer (T5)": {
@@ -23,7 +23,7 @@ def generate_text(model_choice, input_text, max_tokens, temperature, top_p):
     model_info = models[model_choice]
     tokenizer = model_info["tokenizer"]
     model = model_info["model"]
-    
+
     inputs = tokenizer(input_text, return_tensors="pt", padding=True, truncation=True, max_length=512)
     outputs = model.generate(
         **inputs, max_length=max_tokens, num_beams=5, early_stopping=True, temperature=temperature, top_p=top_p