anaspro commited on
Commit
b564e39
·
verified ·
1 Parent(s): 9050d85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "anaspro/Shako-4B-it-v1"
16
  processor = AutoProcessor.from_pretrained(model_id)
17
  model = AutoModelForImageTextToText.from_pretrained(
18
  model_id,
@@ -203,7 +203,7 @@ def generate(message: dict, history: list[dict], system_prompt: str = "", max_ne
203
 
204
  # Examples for the chat interface (with additional inputs: system_prompt, max_new_tokens)
205
  examples = [
206
- ["What is the capital of France?", "You are a helpful assistant.", 700],
207
  ["Explain quantum computing in simple terms", "You are a helpful assistant.", 512],
208
  ["Write a short story about a robot learning to paint", "You are a helpful assistant.", 1000]
209
  ]
 
12
  from transformers.generation.streamers import TextIteratorStreamer
13
 
14
  # Model configuration
15
+ model_id = "anaspro/Shako-4B-it-v2"
16
  processor = AutoProcessor.from_pretrained(model_id)
17
  model = AutoModelForImageTextToText.from_pretrained(
18
  model_id,
 
203
 
204
  # Examples for the chat interface (with additional inputs: system_prompt, max_new_tokens)
205
  examples = [
206
+ ["انت موديل عراقي تحكي هعراقي فقط وتكون ترفيهي", 700],
207
  ["Explain quantum computing in simple terms", "You are a helpful assistant.", 512],
208
  ["Write a short story about a robot learning to paint", "You are a helpful assistant.", 1000]
209
  ]