temp remove pix2tikz

webui.py

@@ -17,7 +17,7 @@ import gradio as gr
 from transformers import TextIteratorStreamer, pipeline, ImageToTextPipeline, AutoModelForPreTraining, AutoProcessor
 
 import os
-from pix2tex.cli import LatexOCR
+# from pix2tex.cli import LatexOCR
 from munch import Munch
 import spaces
 
@@ -26,7 +26,7 @@ from infer import TikzDocument, TikzGenerator
 
 # assets = files(__package__) / "assets" if __package__ else files("assets") / "."
 models = {
-    "pix2tikz": "pix2tikz/mixed_e362_step201.pth",
+    # "pix2tikz": "pix2tikz/mixed_e362_step201.pth",
     "llava-1.5-7b-hf": "waleko/TikZ-llava-1.5-7b",
     "new llava-1.5-7b-hf": "waleko/TikZ-llava-1.5-7b v2"
 }
@@ -57,29 +57,29 @@ def convert_to_svg(pdf):
     return doc[0].get_svg_image()
 
 
-def pix2tikz(
-    checkpoint: str,
-    image: Image.Image,
-    temperature: float,
-    _: float,
-    __: int,
-    ___: bool,
-):
-    cur_pwd = os.path.dirname(os.path.abspath(__file__))
-    config_path = os.path.join(cur_pwd, 'pix2tikz/config.yaml')
-    model_path = os.path.join(cur_pwd, checkpoint)
-
-    print(cur_pwd, config_path, model_path, os.path.exists(config_path), os.path.exists(model_path))
-
-    args = Munch({'config': config_path,
-                  'checkpoint': model_path,
-                  'no_resize': False,
-                  'no_cuda': False,
-                  'temperature': temperature})
-    model = LatexOCR(args)
-    res = model(image)
-    text = re.sub(r'\\n(?=\W)', '\n', res)
-    return text, None, True
+# def pix2tikz(
+#     checkpoint: str,
+#     image: Image.Image,
+#     temperature: float,
+#     _: float,
+#     __: int,
+#     ___: bool,
+# ):
+#     cur_pwd = os.path.dirname(os.path.abspath(__file__))
+#     config_path = os.path.join(cur_pwd, 'pix2tikz/config.yaml')
+#     model_path = os.path.join(cur_pwd, checkpoint)
+#
+#     print(cur_pwd, config_path, model_path, os.path.exists(config_path), os.path.exists(model_path))
+#
+#     args = Munch({'config': config_path,
+#                   'checkpoint': model_path,
+#                   'no_resize': False,
+#                   'no_cuda': False,
+#                   'temperature': temperature})
+#     model = LatexOCR(args)
+#     res = model(image)
+#     text = re.sub(r'\\n(?=\W)', '\n', res)
+#     return text, None, True
 
 
 def inference(
@@ -93,7 +93,7 @@ def inference(
     try:
         image = image_dict['composite']
         if "pix2tikz" in model_name:
-            yield pix2tikz(model_name, image, temperature, top_p, top_k, expand_to_square)
+            # yield pix2tikz(model_name, image, temperature, top_p, top_k, expand_to_square)
             return
 
         generate = TikzGenerator(