Update app.py

app.py

@@ -12,13 +12,13 @@ os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 accelerator = Accelerator()
 
 gpt2_pipeline = accelerator.prepare(
-    pipeline("text-generation", model="Qwen/Qwen-1_8B-Chat", device=accelerator.device)
+    pipeline("text-generation", model="Qwen/Qwen-1_8B-Chat", device=accelerator.device , trust_remote_code=True)
 )
 Najeb_pipeline = accelerator.prepare(
-    pipeline("text-generation", model="najeebjust/Najeeb", device=accelerator.device)
+    pipeline("text-generation", model="najeebjust/Najeeb", device=accelerator.device , trust_remote_code=True)
 )
 llama2_pipeline = accelerator.prepare(
-    pipeline("text-generation", model="Harikrishnan46624/finetuned_llama2-1.1b-chat", device=accelerator.device)
+    pipeline("text-generation", model="Harikrishnan46624/finetuned_llama2-1.1b-chat", device=accelerator.device , trust_remote_code=True)
 )
 '''
 gpt2_pipeline = pipeline("text-generation", model="Qwen/Qwen-1_8B-Chat", device=0 if torch.cuda.is_available() else -1, trust_remote_code=True)