fix webui

infer.py

@@ -196,7 +196,7 @@ class TikzGenerator:
         self.expand_to_square = expand_to_square
         self.clean_up_output = clean_up_output
         self.pipeline = pipe
-        self.pipeline.model = torch.compile(model) # type: ignore
+        # self.pipeline.model = torch.compile(self.pipeline.model)
 
         self.default_kwargs = dict(
             temperature=temperature,

requirements.txt

@@ -4,4 +4,4 @@ pdf2image~=1.16
 PyMuPDF~=1.22
 peft>=0.2.0
 transformers
-gradio~=3.46
+gradio

webui.py

@@ -139,7 +139,8 @@ def build_ui(model=list(models)[0], lock=False, rasterize=False, force_light=Fal
                     "while simple objects work best with shorter, 2-3 word captions."
                 )
                 # caption = gr.Textbox(label="Caption", info=info, placeholder="Type a caption...")
-                image = gr.Image(label="Image Input", type="pil")
+                # image = gr.Image(label="Image Input", type="pil")
+                image = gr.ImageEditor(label="Image Input", type="pil")
                 label = "Model" + (f" ({lock_reason})" if lock else "")
                 model = gr.Dropdown(label=label, choices=list(models.items()), value=models[model], interactive=not lock) # type: ignore
                 with gr.Accordion(label="Advanced Options", open=False):