Update app.py

app.py

@@ -42,7 +42,7 @@ def setup():
             img_pipe.unet.up_blocks[i] = torch.compile(img_pipe.unet.up_blocks[i])
     tokenizer = AutoTokenizer.from_pretrained(model_type, max_length=1024)
     eos_id = tokenizer.encode(tokenizer.eos_token)[0]
-    @spaces.GPU
+    
     def forward_encoder(latex):
         device = ("cuda" if torch.cuda.is_available() else "cpu")
         img_pipe.to(device)
@@ -76,6 +76,8 @@ with gr.Blocks() as demo:
             image = gr.Image(label="Rendered Image", show_label=False, elem_id="image")
     inputs = [textbox]
     outputs = [slider, image, submit_btn]
+
+    @spaces.GPU
     def infer(formula):
         current_time = datetime.now(tz)
         print (current_time, formula)