Update app.py

app.py

@@ -139,6 +139,11 @@ model = AutoModelForCausalLM.from_pretrained(
     attn_implementation="flash_attention_2",
 )
 
+accelerator = Accelerator()
+
+model = accelerator.prepare(model)
+
+
 
 
 import json