재업로드

app/main.py

@@ -1,8 +1,10 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
+
 from fastapi import FastAPI, Request
 from transformers import pipeline
 
 print("✅ FastAPI 시작됨!")
-
 app = FastAPI()
 
 print("📦 모델 로딩 시작")
@@ -20,6 +22,5 @@ async def generate_text(request: Request):
     print(f"📥 프롬프트 입력: {prompt}")
     if not prompt:
         return {"error": "Missing 'prompt' field"}
-
     result = generator(prompt, max_length=50, do_sample=True)
     return {"generated_text": result[0]["generated_text"]}