Upload main.py
Browse files
main.py
CHANGED
|
@@ -6,6 +6,7 @@ from fastapi.responses import Response
|
|
| 6 |
from pydantic import BaseModel
|
| 7 |
from transformers import pipeline
|
| 8 |
|
|
|
|
| 9 |
# Utw贸rz instancj臋 FastAPI
|
| 10 |
app = FastAPI(
|
| 11 |
title="Bielik Text Generation API",
|
|
@@ -33,6 +34,7 @@ async def startup_event():
|
|
| 33 |
To zajmie troch臋 czasu, ale dzieje si臋 tylko raz.
|
| 34 |
"""
|
| 35 |
global generator
|
|
|
|
| 36 |
print(f"艁adowanie modelu: {MODEL_NAME}...")
|
| 37 |
try:
|
| 38 |
# Mo偶esz dostosowa膰 device=0 (GPU) lub device=-1 (CPU) w zale偶no艣ci od wybranej maszyny Space
|
|
|
|
| 6 |
from pydantic import BaseModel
|
| 7 |
from transformers import pipeline
|
| 8 |
|
| 9 |
+
|
| 10 |
# Utw贸rz instancj臋 FastAPI
|
| 11 |
app = FastAPI(
|
| 12 |
title="Bielik Text Generation API",
|
|
|
|
| 34 |
To zajmie troch臋 czasu, ale dzieje si臋 tylko raz.
|
| 35 |
"""
|
| 36 |
global generator
|
| 37 |
+
|
| 38 |
print(f"艁adowanie modelu: {MODEL_NAME}...")
|
| 39 |
try:
|
| 40 |
# Mo偶esz dostosowa膰 device=0 (GPU) lub device=-1 (CPU) w zale偶no艣ci od wybranej maszyny Space
|