jayjay-12345 commited on
Commit
bf432d6
·
verified ·
1 Parent(s): 5885a0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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",