Upload main.py

main.py

@@ -12,7 +12,7 @@ app = FastAPI(
 )
 
 # Ścieżka do modelu - Hugging Face automatycznie pobierze model
-MODEL_NAME = "PL-MoD/Bielik-1.5B-v3.0-Instruct"
+MODEL_NAME = "speakleash/Bielik-1.5B-v3.0-Instruct"
 generator = None # Zostanie załadowany później
 
 # Model wejściowy dla POST request
@@ -59,6 +59,7 @@ async def generate_text(request: GenerationRequest):
     """
     Endpoint do generowania tekstu na podstawie promptu.
     """
+    print(request)
     if generator is None:
         raise HTTPException(status_code=503, detail="Model nie został załadowany lub wystąpił błąd.")
 
@@ -68,7 +69,7 @@ async def generate_text(request: GenerationRequest):
             max_new_tokens=request.max_new_tokens,
             temperature=request.temperature,
             top_p=request.top_p,
-            do_sample=True # Ważne dla generowania z temperaturą
+            do_sample=True, # Ważne dla generowania z temperaturą
         )
         # Pipeline zwraca listę słowników, bierzemy pierwszy wynik
         return {"generated_text": generated_text[0]["generated_text"]}