change branding

webui.py

@@ -33,7 +33,7 @@ def cached_load(model_name, **kwargs) -> ImageToTextPipeline:
     if not is_quantization(model_name):
         return pipeline("image-to-text", model=model_name, **kwargs)
     else:
-        model = AutoModelForPreTraining.from_pretrained(model_name, load_in_8bit=True, **kwargs)
+        model = AutoModelForPreTraining.from_pretrained(model_name, load_in_4bit=True, **kwargs)
         processor = AutoProcessor.from_pretrained(model_name)
         return pipeline(task="image-to-text", model=model, tokenizer=processor.tokenizer, image_processor=processor.image_processor)
 
@@ -107,15 +107,16 @@ def get_banner():
     # Ti*k*Z Assistant: Sketches to Vector Graphics with Ti*k*Z
 
     <p>
-      # <a style="display:inline-block" href="https://github.com/potamides/AutomaTikZ">
-      #   <img src="https://img.shields.io/badge/View%20on%20GitHub-green?logo=github&labelColor=gray" alt="View on GitHub">
-      # </a>
-      # <a style="display:inline-block" href="https://arxiv.org/abs/2310.00367">
-      #   <img src="https://img.shields.io/badge/View%20on%20arXiv-B31B1B?logo=arxiv&labelColor=gray" alt="View on arXiv">
-      # </a>
-      # <a style="display:inline-block" href="https://colab.research.google.com/drive/14S22x_8VohMr9pbnlkB4FqtF4n81khIh">
-      #   <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open in Colab">
-      # </a>
+    
+      <!--<a style="display:inline-block" href="https://github.com/potamides/AutomaTikZ">
+        <img src="https://img.shields.io/badge/View%20on%20GitHub-green?logo=github&labelColor=gray" alt="View on GitHub">
+      </a>
+      <a style="display:inline-block" href="https://arxiv.org/abs/2310.00367">
+        <img src="https://img.shields.io/badge/View%20on%20arXiv-B31B1B?logo=arxiv&labelColor=gray" alt="View on arXiv">
+      </a>
+      <a style="display:inline-block" href="https://colab.research.google.com/drive/14S22x_8VohMr9pbnlkB4FqtF4n81khIh">
+        <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open in Colab">
+      </a>-->
       <a style="display:inline-block" href="https://huggingface.co/spaces/waleko/TikZ-Assistant">
         <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/open-in-hf-spaces-sm.svg" alt="Open in HF Spaces">
       </a>