adyaprasad commited on
Commit
cc17b1b
·
verified ·
1 Parent(s): bd4ada0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ def home():
18
  def generate(text: str):
19
  # use the pipeline to generate text from given input text
20
  output = pipe(text, max_new_tokens=50, temperature=0.7, top_k=50) # Adjust parameters as needed
 
21
  #return the correct key based on model output
22
  return {"output": output[0]['generated_text']}
23
 
 
18
  def generate(text: str):
19
  # use the pipeline to generate text from given input text
20
  output = pipe(text, max_new_tokens=50, temperature=0.7, top_k=50) # Adjust parameters as needed
21
+ print(output) # Debugging line
22
  #return the correct key based on model output
23
  return {"output": output[0]['generated_text']}
24