Georg4000 commited on
Commit
2232874
ยท
verified ยท
1 Parent(s): c6f760b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ else:
14
  raise ValueError("Hugging Face token is missing. Please check your secrets.")
15
 
16
  # ุงุณุชุฎุฏุงู… ู†ู…ูˆุฐุฌ TinyLlama ุงู„ุฃุฎู
17
- model_name = "TinyLlama/TinyLlama-1.1B-Chat"
18
 
19
  # ุชุญุฏูŠุฏ ุงู„ุฌู‡ุงุฒ ุงู„ู…ู†ุงุณุจ (GPU ุฃูˆ CPU)
20
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
14
  raise ValueError("Hugging Face token is missing. Please check your secrets.")
15
 
16
  # ุงุณุชุฎุฏุงู… ู†ู…ูˆุฐุฌ TinyLlama ุงู„ุฃุฎู
17
+ model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
18
 
19
  # ุชุญุฏูŠุฏ ุงู„ุฌู‡ุงุฒ ุงู„ู…ู†ุงุณุจ (GPU ุฃูˆ CPU)
20
  device = "cuda" if torch.cuda.is_available() else "cpu"