updated

app.py

@@ -3,11 +3,12 @@
 #
 import spaces
 from diffusers import DiffusionPipeline
-
+import os
 import torch
 from transformers import pipeline
 
-pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_dtype=torch.bfloat16, device_map="auto", use_auth_token=True)
+token = os.getenv("HUGGINGFACE_API_TOKEN")
+pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_dtype=torch.bfloat16, device_map="auto", use_auth_token=token)
 response = pipe(chat, max_new_tokens=512)
 
 pipe.to('cuda')