Spaces:

nero7151
/

SPOON-AI

Sleeping

nero7151

재업로드

34bcb90 over 1 year ago

764 Bytes

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=100,
    do_sample=True,         
    temperature=0.8,        
    top_k=50               
)
    return {"generated_text": result[0]["generated_text"]}

	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=100,
	do_sample=True,
	temperature=0.8,
	top_k=50
	)
	return {"generated_text": result[0]["generated_text"]}