Navneetkumar11 commited on
Commit
c53fe55
·
verified ·
1 Parent(s): 97a1c4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,10 +14,11 @@ def create_compliment(text_description: str) -> str:
14
  prompt = f"Please turn the following description into a short, fun, and friendly compliment: {text_description}"
15
 
16
  # We send the prompt to the language model to get our final result.
 
17
  compliment = llm_client.predict(
18
  prompt=prompt,
19
  max_new_tokens=128,
20
- api_name="/add_text"
21
  )
22
 
23
  return compliment
 
14
  prompt = f"Please turn the following description into a short, fun, and friendly compliment: {text_description}"
15
 
16
  # We send the prompt to the language model to get our final result.
17
+ # CORRECTED api_name from "/add_text" to "/predict"
18
  compliment = llm_client.predict(
19
  prompt=prompt,
20
  max_new_tokens=128,
21
+ api_name="/predict"
22
  )
23
 
24
  return compliment