infinityy commited on
Commit
d9f64f4
·
verified ·
1 Parent(s): ca4e3d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -99,7 +99,7 @@ def suggest_title(component, description,pid):
99
  generated_title = generate_title(soft_prompt, new_doc_text)
100
  return f"{generated_title}"
101
 
102
- def infer(component,title,description,pid=None,mode="priority"):
103
  if mode=="title":
104
  return suggest_title(component, description,pid)
105
  # BERT embedding
 
99
  generated_title = generate_title(soft_prompt, new_doc_text)
100
  return f"{generated_title}"
101
 
102
+ def infer(component,title="",description="",pid=None,mode="priority"):
103
  if mode=="title":
104
  return suggest_title(component, description,pid)
105
  # BERT embedding