Update app.py

app.py

@@ -1,9 +1,15 @@
-# Import the necessary library for creating the interface
+# Import the necessary libraries for creating the interface
 import gradio as gr
-
-# Import the pipeline function from the transformers library to load the model
 from transformers import pipeline
 
+# Verify model loading
+def verify_model():
+    try:
+        gif_generator = pipeline('text-to-video', model='hotshotco/Hotshot-XL')
+        return "Model loaded successfully!"
+    except Exception as e:
+        return str(e)
+
 # Define a function to apply regex choices to the prompt text
 def apply_regex_choices(text):
     import re  # Import the regular expressions module
@@ -31,8 +37,8 @@ prompt_template = """
 Create a GIF of a 'Technical Difficulties' television program interruption. The scene should show colored glitchy and typical 'technical difficulties' cable interruptions, (using|with) demonic symbols glitching out along with the colors of the usual 'tech difficulties' broadcasts. The setting should be (an older|a vintage) tv cable broadcast signal interruption. The style should be a realistic cable interruption interlaced with (demonic imagery|dark symbols) glitching out along with the typical color page of interruption in connection. The GIF should last for 1 second.
 """
 
-# Create a Gradio interface
-iface = gr.Interface(
+# Create the main Gradio interface
+main_iface = gr.Interface(
     fn=generate_gif,  # The function to call when the user interacts with the interface
     inputs=gr.Textbox(lines=10, placeholder="Enter your custom prompt here or use the default template", value=prompt_template),  # Input textbox for the prompt
     outputs="file",  # Output will be a file (the generated GIF)
@@ -44,6 +50,23 @@ iface = gr.Interface(
     ]
 )
 
+# Create the model verification interface
+test_iface = gr.Interface(
+    fn=verify_model,
+    inputs=None,
+    outputs="text",
+    title="Test Model Loading",
+    description="Click the button to test if the model loads correctly."
+)
+
+# Combine both interfaces into a single Gradio Block
+block = gr.Blocks()
+with block:
+    gr.Markdown("# Structured Prompt GIF Generator")
+    gr.Tab("Generate GIFs", main_iface)
+    gr.Tab("Verify Model", test_iface)
+
 # Launch the Gradio interface
-iface.launch()
+block.launch()
+