M Farrel Danendra Rachim commited on
Commit
c1077b5
·
1 Parent(s): 7744e9e

feat: add sentence example

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -53,7 +53,7 @@ async def emotion(request: Request):
53
  async def sentence_example(request: Request):
54
  try:
55
  prompt = f"Generate a sentence using the word: {request.text}"
56
- result = text2text_generator(prompt)
57
  return {"text": result[0]['generated_text']}
58
  except Exception as e:
59
  return {"text": f"Error: {str(e)}"}
 
53
  async def sentence_example(request: Request):
54
  try:
55
  prompt = f"Generate a sentence using the word: {request.text}"
56
+ result = text2text_generator(prompt, max_length=10)
57
  return {"text": result[0]['generated_text']}
58
  except Exception as e:
59
  return {"text": f"Error: {str(e)}"}