artelo1 commited on
Commit
9f62497
verified
1 Parent(s): 23ec108

Upload main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -0
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