Update app.py
Browse files
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
|