Adir Gozlan
commited on
Commit
•
f90eee6
1
Parent(s):
b596009
fix cross encoder None
Browse files
app.py
CHANGED
@@ -111,7 +111,7 @@ with gr.Blocks() as demo:
|
|
111 |
txt_btn = gr.Button(value="Submit text", scale=1)
|
112 |
|
113 |
api_kind = gr.Radio(choices=["HuggingFace", "OpenAI"], value="HuggingFace", label="LLM")
|
114 |
-
cross_enc = gr.Radio(choices=[
|
115 |
|
116 |
prompt_html = gr.HTML()
|
117 |
# Turn off interactivity while generating if you click
|
|
|
111 |
txt_btn = gr.Button(value="Submit text", scale=1)
|
112 |
|
113 |
api_kind = gr.Radio(choices=["HuggingFace", "OpenAI"], value="HuggingFace", label="LLM")
|
114 |
+
cross_enc = gr.Radio(choices=[None, "cross-encoder/ms-marco-MiniLM-L-6-v2", "BAAI/bge-reranker-large"], value=None, label="Cross Encoder")
|
115 |
|
116 |
prompt_html = gr.HTML()
|
117 |
# Turn off interactivity while generating if you click
|