Update app.py

app.py

@@ -356,7 +356,7 @@ import torch.nn.functional as F
 from sageattention import sageattn
 import random
 import uuid
-#import gradio as gr
+import gradio as gr
 import numpy as np
 from PIL import Image
 
@@ -760,7 +760,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             run_button_30c.click,
         ],
       #  api_name="generate",  # Add this line
-        fn=generate_30c,
+        fn=pyx.generate_30c,
         inputs=[
             prompt,
             negative_prompt,