Update app.py
Browse files
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
|
18 |
|
19 |
# Gunakan model open source yang tidak memerlukan login
|
20 |
-
model_id = "
|
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
|