new name and finishes

README.md

@@ -1,5 +1,5 @@
 ---
-title: Img2html
+title: Screenshot to HTML
 emoji: ⚡
 colorFrom: red
 colorTo: green
@@ -9,4 +9,4 @@ app_file: app.py
 pinned: false
 ---
 
-Demo
+Screenshot to HTML/CSS demo.

app.py

@@ -17,11 +17,11 @@ from transformers.image_transforms import resize, to_channel_dimension_format
 API_TOKEN = os.getenv("HF_AUTH_TOKEN")
 DEVICE = torch.device("cuda")
 PROCESSOR = AutoProcessor.from_pretrained(
-    "HuggingFaceM4/img2html",
+    "HuggingFaceM4/VLM_WebSight_finetuned",
     token=API_TOKEN,
 )
 MODEL = AutoModelForCausalLM.from_pretrained(
-    "HuggingFaceM4/img2html",
+    "HuggingFaceM4/VLM_WebSight_finetuned",
     token=API_TOKEN,
     trust_remote_code=True,
     torch_dtype=torch.bfloat16,

requirements.txt

@@ -3,3 +3,4 @@ transformers
 torch
 packaging
 ninja
+einops