Upload 4 files

app/main.py

@@ -1,7 +1,22 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
+from transformers import pipeline
+
+# 모델 로드
+generator = pipeline("text-generation", model="gpt2")
 
 app = FastAPI()
 
 @app.get("/")
-def read_root():
-    return {"message": "✅ Hello from FastAPI in Docker Space!"}
+def root():
+    return {"message": "✅ FastAPI with Hugging Face GPT2!"}
+
+@app.post("/generate")
+async def generate_text(request: Request):
+    body = await request.json()
+    prompt = body.get("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"]}

requirements.txt

@@ -1,2 +1,4 @@
 fastapi
-uvicorn
+uvicorn[standard]
+transformers
+torch