try fix zerogpu

webui.py

@@ -19,6 +19,7 @@ from transformers import TextIteratorStreamer, pipeline, ImageToTextPipeline, Au
 import os
 from pix2tex.cli import LatexOCR
 from munch import Munch
+import spaces
 
 
 from infer import TikzDocument, TikzGenerator
@@ -109,7 +110,7 @@ def inference(
         )
 
         thread = ThreadPool(processes=1)
-        async_result = thread.apply_async(generate, kwds=dict(image=image, streamer=streamer))
+        async_result = thread.apply_async(spaces.GPU(generate), kwds=dict(image=image, streamer=streamer))
         generated_text = ""
         for new_text in streamer:
             generated_text += new_text