Arifzyn commited on
Commit
e713c44
·
verified ·
1 Parent(s): b8a7d1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,10 +14,10 @@ logging.basicConfig(
14
  )
15
  logger = logging.getLogger(__name__)
16
 
17
- app = FastAPI(title="TinyLlama API", description="API untuk model TinyLlama Chat")
18
 
19
  # Gunakan model open source yang tidak memerlukan login
20
- model_id = "microsoft/phi-1_5" # Model ringan untuk free tier HF
21
  model_dir = "model_cache" # Direktori untuk menyimpan model
22
 
23
  # Variabel global untuk menyimpan model dan tokenizer
 
14
  )
15
  logger = logging.getLogger(__name__)
16
 
17
+ app = FastAPI(title="TinyLlama API", description="API untuk model TinyLlama-1.1B-Chat")
18
 
19
  # Gunakan model open source yang tidak memerlukan login
20
+ model_id = "TinyLlama/TinyLlama-1.1B-Chat-v1.0" # Model TinyLlama Chat
21
  model_dir = "model_cache" # Direktori untuk menyimpan model
22
 
23
  # Variabel global untuk menyimpan model dan tokenizer