Upload 3 files

Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.10
+
+WORKDIR /app
+
+COPY . /app
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app/main.py

@@ -0,0 +1,25 @@
+from fastapi import FastAPI, Request
+from transformers import pipeline
+
+print("✅ FastAPI 시작됨!")
+
+app = FastAPI()
+
+print("📦 모델 로딩 시작")
+generator = pipeline("text-generation", model="gpt2")
+print("✅ 모델 로딩 완료!")
+
+@app.get("/")
+def root():
+    return {"message": "✅ FastAPI 정상 작동 중!"}
+
+@app.post("/generate")
+async def generate_text(request: Request):
+    body = await request.json()
+    prompt = body.get("prompt", "")
+    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"]}

requirements.txt

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