Update app.py

app.py

@@ -10,7 +10,13 @@ from transformers import pipeline
 import torch
 
 # ---------------- CONFIG ----------------
-MODEL_ID = "openai/oss/20b"
+MODEL_ID = "openai/gpt-oss-20b"
+pipe = pipeline(
+    "text-generation",
+    model=model_id,
+    torch_dtype="auto",
+    device_map="auto",
+)
 DATA_DIR = "/data" if os.path.isdir("/data") else "./data"
 os.makedirs(DATA_DIR, exist_ok=True)