Update app.py

app.py

@@ -2,6 +2,10 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from transformers import pipeline, set_seed
+import os
+
+# Set cache model ke folder yang bisa ditulis
+os.environ["TRANSFORMERS_CACHE"] = "/app/.cache"
 
 app = FastAPI(title="StemdjaGPT", version="1.0")
 
@@ -14,13 +18,12 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Load model GPT Indonesia
+# Load model GPT
 MODEL = "cahya/gpt2-small-indonesian-522M"
 generator = pipeline("text-generation", model=MODEL, tokenizer=MODEL)
 
 set_seed(42)
 
-# Input schema
 class PromptInput(BaseModel):
     prompt: str
     max_length: int = 100