api2

Runtime error

Reality123b commited on Oct 2, 2025

Commit

3003e89

verified ·

1 Parent(s): 731d6b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ app = FastAPI()
 # Initialize text generation pipeline
 generator = pipeline(
     "text-generation",
-    model="HuggingFaceTB/SmolLM2-360M",
     device="cpu"  # Use CPU (change to device=0 for GPU)
 )

 # Initialize text generation pipeline
 generator = pipeline(
     "text-generation",
+    model="HuggingFaceTB/SmolLM2-360M-Instruct",
     device="cpu"  # Use CPU (change to device=0 for GPU)
 )