Update app.py
Browse files
app.py
CHANGED
|
@@ -65,9 +65,9 @@ with open("processed_chunks.json", "r") as f:
|
|
| 65 |
# (Using Path 1: Ungated models to avoid the 401 Unauthorized Error)
|
| 66 |
MODELS = {
|
| 67 |
"Granite 3.1 2B": "ibm-granite/granite-3.1-2b-instruct",
|
| 68 |
-
"Phi 3.5 Mini": "microsoft/Phi-3.5-mini-instruct",
|
| 69 |
"Qwen 2.5 1.5B": "Qwen/Qwen2.5-1.5B-Instruct",
|
| 70 |
-
"SmolLM 1.7B": "HuggingFaceTB/SmolLM-1.7B-Instruct"
|
|
|
|
| 71 |
}
|
| 72 |
|
| 73 |
def ask_specific_model(model_name, prompt):
|
|
@@ -120,9 +120,9 @@ interface = gr.Interface(
|
|
| 120 |
inputs=gr.Textbox(label="Ask an HR Question"),
|
| 121 |
outputs=[
|
| 122 |
gr.Textbox(label="IBM Granite 3.1"),
|
|
|
|
|
|
|
| 123 |
gr.Textbox(label="Microsoft Phi 3.5"),
|
| 124 |
-
gr.Textbox(label="Qwen 2.5 1.5B"),
|
| 125 |
-
gr.Textbox(label="SmolLM 1.7B"),
|
| 126 |
gr.Textbox(label="Source Used")
|
| 127 |
],
|
| 128 |
title="RAG Model Benchmarking",
|
|
|
|
| 65 |
# (Using Path 1: Ungated models to avoid the 401 Unauthorized Error)
|
| 66 |
MODELS = {
|
| 67 |
"Granite 3.1 2B": "ibm-granite/granite-3.1-2b-instruct",
|
|
|
|
| 68 |
"Qwen 2.5 1.5B": "Qwen/Qwen2.5-1.5B-Instruct",
|
| 69 |
+
"SmolLM 1.7B": "HuggingFaceTB/SmolLM-1.7B-Instruct",
|
| 70 |
+
"Phi 3.5 Mini": "microsoft/Phi-3.5-mini-instruct",
|
| 71 |
}
|
| 72 |
|
| 73 |
def ask_specific_model(model_name, prompt):
|
|
|
|
| 120 |
inputs=gr.Textbox(label="Ask an HR Question"),
|
| 121 |
outputs=[
|
| 122 |
gr.Textbox(label="IBM Granite 3.1"),
|
| 123 |
+
gr.Textbox(label="Alibaba Qwen 2.5 1.5B"),
|
| 124 |
+
gr.Textbox(label="HuggingFaceTB SmolLM 1.7B"),
|
| 125 |
gr.Textbox(label="Microsoft Phi 3.5"),
|
|
|
|
|
|
|
| 126 |
gr.Textbox(label="Source Used")
|
| 127 |
],
|
| 128 |
title="RAG Model Benchmarking",
|