Spaces:
Sleeping
Sleeping
nero7151 commited on
Commit ·
ca84447
1
Parent(s): 9fa6a6b
재업로드
Browse files- app/__pycache__/main.cpython-313.pyc +0 -0
- app/main.py +7 -1
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
|
@@ -19,5 +19,11 @@ async def generate_text(request: Request):
|
|
| 19 |
print(f"📥 프롬프트 입력: {prompt}")
|
| 20 |
if not prompt:
|
| 21 |
return {"error": "Missing 'prompt' field"}
|
| 22 |
-
result = generator(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
return {"generated_text": result[0]["generated_text"]}
|
|
|
|
| 19 |
print(f"📥 프롬프트 입력: {prompt}")
|
| 20 |
if not prompt:
|
| 21 |
return {"error": "Missing 'prompt' field"}
|
| 22 |
+
result = generator(
|
| 23 |
+
prompt,
|
| 24 |
+
max_length=100,
|
| 25 |
+
do_sample=True,
|
| 26 |
+
temperature=0.8,
|
| 27 |
+
top_k=50
|
| 28 |
+
)
|
| 29 |
return {"generated_text": result[0]["generated_text"]}
|