Update app.py

app.py

@@ -1,8 +1,12 @@
 import gradio as gr
 from diffusers import DiffusionPipeline
+from transformers import AutoTokenizer
+
+# Load the tokenizer
+tokenizer = AutoTokenizer.from_pretrained("./")
 
 # Load the pre-trained model
-pipe = DiffusionPipeline.from_pretrained("rain1011/pyramid-flow-miniflux")
+pipe = DiffusionPipeline.from_pretrained("./", tokenizer=tokenizer)
 
 # Function to generate video from text prompt
 def generate_video(prompt):