Update app.py

app.py

@@ -6,13 +6,20 @@ from PIL import Image
 import torch
 import os, time
 
+# Load the model and processor
+# model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
+#     "daniel3303/QwenStoryteller",
+#     torch_dtype=torch.float16,
+#     device_map="auto"
+# )
+# processor = AutoProcessor.from_pretrained("daniel3303/QwenStoryteller")
 # Load the model and processor
 model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
-    "daniel3303/QwenStoryteller",
+    "daniel3303/QwenStoryteller2",
     torch_dtype=torch.float16,
     device_map="auto"
 )
-processor = AutoProcessor.from_pretrained("daniel3303/QwenStoryteller")
+processor = AutoProcessor.from_pretrained("daniel3303/QwenStoryteller2")
 
 @spaces.GPU()
 @torch.no_grad()
@@ -107,7 +114,7 @@ with gr.Blocks(fill_height=True) as demo:
             * Grounded Storytelling: Uses specialized XML tags to link narrative elements directly to visual entities
             * Reduced Hallucinations: Achieves 12.3% fewer hallucinations compared to the non-fine-tuned base model
             
-            Model trained by daniel3303, [repository here.](https://huggingface.co/daniel3303/QwenStoryteller)
+            Model trained by daniel3303, [repository here.](https://huggingface.co/daniel3303/QwenStoryteller2)
             """    
         )