updated

app.py

@@ -6,6 +6,7 @@ from diffusers import DiffusionPipeline
 import os
 import torch
 from transformers import pipeline
+import gradio as gr
 
 token = os.getenv("HUGGINGFACE_API_TOKEN")
 pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_dtype=torch.bfloat16, device_map="auto", token=token)