Hugging Face's logo

Spaces:
llamafactory
/
PaliGemma-3B-Chat-v0.2
Running on Zero

App Files Community
1
hiyouga commited on
Commit
e67e492
•
1 Parent(s): 2ca3c06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,9 +8,9 @@ from PIL import Image
8
  from transformers import AutoModelForVision2Seq, AutoProcessor, AutoTokenizer, TextIteratorStreamer
9
 
10
 
11
- TITLE = "<h1><center>Chat with PaliGemma-3B-Chat-v0.1</center></h1>"
12
 
13
- DESCRIPTION = "<h3><center>Visit <a href='https://huggingface.co/hiyouga/PaliGemma-3B-Chat-v0.1' target='_blank'>our model page</a> for details.</center></h3>"
14
 
15
  CSS = """
16
  .duplicate-button {
@@ -22,7 +22,7 @@ CSS = """
22
  """
23
 
24
 
25
- model_id = "hiyouga/PaliGemma-3B-Chat-v0.1"
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)
27
  processor = AutoProcessor.from_pretrained(model_id)
28
  model = AutoModelForVision2Seq.from_pretrained(model_id, torch_dtype="auto", device_map="auto")
 
8
  from transformers import AutoModelForVision2Seq, AutoProcessor, AutoTokenizer, TextIteratorStreamer
9
 
10
 
11
+ TITLE = "<h1><center>Chat with PaliGemma-3B-Chat-v0.2</center></h1>"
12
 
13
+ DESCRIPTION = "<h3><center>Visit <a href='https://huggingface.co/BUAADreamer/PaliGemma-3B-Chat-v0.2' target='_blank'>our model page</a> for details.</center></h3>"
14
 
15
  CSS = """
16
  .duplicate-button {
 
22
  """
23
 
24
 
25
+ model_id = "BUAADreamer/PaliGemma-3B-Chat-v0.2"
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)
27
  processor = AutoProcessor.from_pretrained(model_id)
28
  model = AutoModelForVision2Seq.from_pretrained(model_id, torch_dtype="auto", device_map="auto")