Update app.py

app.py

@@ -1,31 +1,37 @@
 import gradio as gr
-from transformers import AutoProcessor, VisionEncoderDecoderModel
+from transformers import (
+    SiglipImageProcessor,
+    RobertaTokenizerFast,
+    VisionEncoderDecoderModel
+)
 from PIL import Image
 import torch
 
-# 加载模型和处理器
+# 模型 ID
 model_id = "starvector/starvector-8b-im2svg"
-processor = AutoProcessor.from_pretrained(model_id)
+
+# 分别加载 image processor 和 tokenizer
+image_processor = SiglipImageProcessor.from_pretrained(model_id)
+tokenizer = RobertaTokenizerFast.from_pretrained(model_id)
 model = VisionEncoderDecoderModel.from_pretrained(model_id)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
-# 定义推理函数
+# 推理函数
 def im2svg(image):
-    inputs = processor(images=image, return_tensors="pt").to(device)
+    inputs = image_processor(images=image, return_tensors="pt").to(device)
     outputs = model.generate(**inputs, max_new_tokens=1024)
-    generated_svg = processor.batch_decode(outputs, skip_special_tokens=True)[0]
-    return generated_svg
+    svg_code = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
+    return svg_code
 
-# 创建 Gradio 界面
+# Gradio UI
 demo = gr.Interface(
     fn=im2svg,
     inputs=gr.Image(type="pil"),
     outputs="text",
-    title="StarVector 8B - Image to SVG",
-    description="上传图像，将其转化为 SVG 矢量代码。",
-    allow_flagging="never"
+    title="🖼️ StarVector: Image → SVG",
+    description="上传图像，我将它转化为矢量图（SVG 代码）。适用于简笔画、图标、草图。",
 )
 
 demo.launch()