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
|