VictorSanh commited on
Commit
47d2396
1 Parent(s): 7df82fd

new name and finishes

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  2. app.py +2 -2
  3. requirements.txt +1 -0
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Img2html
3
  emoji: ⚡
4
  colorFrom: red
5
  colorTo: green
@@ -9,4 +9,4 @@ app_file: app.py
9
  pinned: false
10
  ---
11
 
12
- Demo
 
1
  ---
2
+ title: Screenshot to HTML
3
  emoji: ⚡
4
  colorFrom: red
5
  colorTo: green
 
9
  pinned: false
10
  ---
11
 
12
+ Screenshot to HTML/CSS demo.
app.py CHANGED
@@ -17,11 +17,11 @@ from transformers.image_transforms import resize, to_channel_dimension_format
17
  API_TOKEN = os.getenv("HF_AUTH_TOKEN")
18
  DEVICE = torch.device("cuda")
19
  PROCESSOR = AutoProcessor.from_pretrained(
20
- "HuggingFaceM4/img2html",
21
  token=API_TOKEN,
22
  )
23
  MODEL = AutoModelForCausalLM.from_pretrained(
24
- "HuggingFaceM4/img2html",
25
  token=API_TOKEN,
26
  trust_remote_code=True,
27
  torch_dtype=torch.bfloat16,
 
17
  API_TOKEN = os.getenv("HF_AUTH_TOKEN")
18
  DEVICE = torch.device("cuda")
19
  PROCESSOR = AutoProcessor.from_pretrained(
20
+ "HuggingFaceM4/VLM_WebSight_finetuned",
21
  token=API_TOKEN,
22
  )
23
  MODEL = AutoModelForCausalLM.from_pretrained(
24
+ "HuggingFaceM4/VLM_WebSight_finetuned",
25
  token=API_TOKEN,
26
  trust_remote_code=True,
27
  torch_dtype=torch.bfloat16,
requirements.txt CHANGED
@@ -3,3 +3,4 @@ transformers
3
  torch
4
  packaging
5
  ninja
 
 
3
  torch
4
  packaging
5
  ninja
6
+ einops