anaspro commited on
Commit
4774881
·
verified ·
1 Parent(s): 68bb543

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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 = "unsloth/gemma-3n-E4B-it"
16
  processor = AutoProcessor.from_pretrained(model_id)
17
  model = AutoModelForImageTextToText.from_pretrained(
18
  model_id,
@@ -203,9 +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
- ["انت موديل عراقي تحكي هعراقي فقط وتكون ترفيهي", 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
  ]
210
 
211
  # Create the chat interface
 
12
  from transformers.generation.streamers import TextIteratorStreamer
13
 
14
  # Model configuration
15
+ model_id = "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
  ]
208
 
209
  # Create the chat interface