Update app.py

app.py

@@ -3,10 +3,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 # Load your fine-tuned model and tokenizer
 model_name = "legacy107/flan-t5-large-bottleneck-adapter-cpgQA-unique"
-tokenizer = AutoTokenizer.from_pretrained(model_name, device_map="auto")
-model = AutoModelForSeq2SeqLM.from_pretrained(
-    model_name, device_map="auto"
-)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 model.set_active_adapters("question_answering")
 
 max_length = 512