anaspro commited on
Commit
3619ce6
·
1 Parent(s): d702c73
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from transformers import AutoModelForImageTextToText, AutoProcessor
12
  from transformers.generation.streamers import TextIteratorStreamer
13
 
14
  # Model configuration
15
- model_id = "google/gemma-7b-it"
16
  processor = AutoProcessor.from_pretrained(model_id)
17
  model = AutoModelForImageTextToText.from_pretrained(
18
  model_id,
 
12
  from transformers.generation.streamers import TextIteratorStreamer
13
 
14
  # Model configuration
15
+ model_id = "anaspro/Shako-4B-it-v5"
16
  processor = AutoProcessor.from_pretrained(model_id)
17
  model = AutoModelForImageTextToText.from_pretrained(
18
  model_id,