Merge branch 'main' of https://huggingface.co/spaces/nero7151/SPOON-AI

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

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"]}

fastapi_gpt2_space/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"]

fastapi_gpt2_space/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"]}

fastapi_gpt2_space/requirements.txt

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

requirements.txt

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