Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import spaces
 from dataclasses import dataclass
 
 import gradio as gr
@@ -731,6 +732,7 @@ print(result)
 # model = Flux().to(dtype=torch.bfloat16, device="cuda")
 # result = model.load_state_dict(load_file("/storage/dev/nyanko/flux-dev/flux1-dev.sft"))
 
+@spaces.GPU
 @torch.inference_mode()
 def generate_image(
     prompt, width, height, guidance, seed,
@@ -852,7 +854,7 @@ def create_demo():
             
             with gr.Column():
                 output_image = gr.Image(label="Generated Image")
-                output_seed = gr.Number(label="Used Seed", precision=0)
+                output_seed = gr.Text(label="Used Seed")
 
         do_img2img.change(
             fn=lambda x: [gr.update(visible=x), gr.update(visible=x), gr.update(visible=x)],