Daemontatox commited on
Commit
a76f205
·
verified ·
1 Parent(s): 03d6372

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -172,16 +172,16 @@ retriever = db.as_retriever(
172
 
173
 
174
  quantization_config = BitsAndBytesConfig(
175
- load_in_8bit=True,
176
- bnb_8bit_compute_dtype=torch.bfloat16,
177
- bnb_8bit_quant_type="nf4",
178
- bnb_8bit_use_double_quant=True
179
  )
180
 
181
 
182
 
183
 
184
- model_id = "mistralai/Mistral-Nemo-Instruct-2407"
185
  tokenizer = AutoTokenizer.from_pretrained(model_id)
186
 
187
  model = AutoModelForCausalLM.from_pretrained(
@@ -280,11 +280,11 @@ def clear_chat():
280
  return [], ""
281
 
282
  # Gradio Interface
283
- with gr.Blocks(theme=gr.themes.Soft()) as iface:
284
  gr.Image("Image.jpg" , width=1200 , height=300 ,show_label=False, show_download_button=False)
285
- gr.Markdown("# Mawared HR Assistant")
286
  gr.Markdown('### Instructions')
287
- gr.Markdown("The first question will always send out an error in chat , try again and the flow should continue normally , its an API issue and we are working on it")
288
 
289
 
290
 
 
172
 
173
 
174
  quantization_config = BitsAndBytesConfig(
175
+ load_in_4bit=True,
176
+ bnb_4bit_compute_dtype=torch.bfloat16,
177
+ bnb_4bit_quant_type="nf4",
178
+ bnb_4bit_use_double_quant=True
179
  )
180
 
181
 
182
 
183
 
184
+ model_id = "Daemontatox/MawaredT3"
185
  tokenizer = AutoTokenizer.from_pretrained(model_id)
186
 
187
  model = AutoModelForCausalLM.from_pretrained(
 
280
  return [], ""
281
 
282
  # Gradio Interface
283
+ with gr.Blocks(theme='lone17/kotaemon') as iface:
284
  gr.Image("Image.jpg" , width=1200 , height=300 ,show_label=False, show_download_button=False)
285
+ gr.Markdown("# Mawared HR Assistant 2.5.1")
286
  gr.Markdown('### Instructions')
287
+ gr.Markdown("Ask a question about MawaredHR and get a detailed answer , if you get an error try again with same prompt , its an Api issue and we are working on it 😀")
288
 
289
 
290