mayf commited on
Commit
c5b69e3
·
verified ·
1 Parent(s): ecd3e31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def load_models():
30
  # Story generation model (Qwen3-1.7B)
31
  storyteller = pipeline(
32
  "text-generation",
33
- model="Qwen/Qwen3-0.5B",
34
  device_map="auto",
35
  trust_remote_code=True,
36
  torch_dtype="auto",
 
30
  # Story generation model (Qwen3-1.7B)
31
  storyteller = pipeline(
32
  "text-generation",
33
+ model="Qwen/Qwen3-0.6B",
34
  device_map="auto",
35
  trust_remote_code=True,
36
  torch_dtype="auto",