Update app.py

app.py

@@ -11,7 +11,8 @@ import base64
 #model_checkpoint = "LaMini-Flan-T5-248M"
 model_checkpoint = "MBZUAI/LaMini-Flan-T5-783M"
 model_tokenizer = T5Tokenizer.from_pretrained(model_checkpoint)
-model = T5ForConditionalGeneration.from_pretrained(model_checkpoint, device_map='auto', torch_dtype=torch.float32)
+#model = T5ForConditionalGeneration.from_pretrained(model_checkpoint, device_map='auto', torch_dtype=torch.float32)
+model = T5ForConditionalGeneration.from_pretrained(model_checkpoint)
 
 #REPO_ID = "MBZUAI/LaMini-Flan-T5-783M"
 #model = pipeline(task='summarization', model=REPO_ID, token=access_token)