nero7151 commited on
Commit
34bcb90
·
1 Parent(s): ca84447

재업로드

Browse files
app/__pycache__/main.cpython-313.pyc CHANGED
Binary files a/app/__pycache__/main.cpython-313.pyc and b/app/__pycache__/main.cpython-313.pyc differ
 
app/main.py CHANGED
@@ -1,22 +1,22 @@
1
  from fastapi import FastAPI, Request
2
  from transformers import pipeline
3
 
4
- print("✅ FastAPI 시작됨!")
5
  app = FastAPI()
6
 
7
- print("📦 모델 로딩 시작")
8
  generator = pipeline("text-generation", model="gpt2")
9
- print("✅ 모델 로딩 완료!")
10
 
11
  @app.get("/")
12
  def root():
13
- return {"message": "✅ FastAPI 정상 작동 중!"}
14
 
15
  @app.post("/generate")
16
  async def generate_text(request: Request):
17
  body = await request.json()
18
  prompt = body.get("prompt", "")
19
- print(f"📥 프롬프트 입력: {prompt}")
20
  if not prompt:
21
  return {"error": "Missing 'prompt' field"}
22
  result = generator(
 
1
  from fastapi import FastAPI, Request
2
  from transformers import pipeline
3
 
4
+ print("FastAPI 시작됨!")
5
  app = FastAPI()
6
 
7
+ print("모델 로딩 시작")
8
  generator = pipeline("text-generation", model="gpt2")
9
+ print("모델 로딩 완료!")
10
 
11
  @app.get("/")
12
  def root():
13
+ return {"message": "FastAPI 정상 작동 중!"}
14
 
15
  @app.post("/generate")
16
  async def generate_text(request: Request):
17
  body = await request.json()
18
  prompt = body.get("prompt", "")
19
+ print(f"프롬프트 입력: {prompt}")
20
  if not prompt:
21
  return {"error": "Missing 'prompt' field"}
22
  result = generator(