Detsutut commited on
Commit
b34f083
1 Parent(s): 18ca0f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ import torch
5
  import re
6
 
7
  # Initialize the model
8
- model = AutoModelForCausalLM.from_pretrained("Detsutut/Igea-1B-qa-GGUF-Q4", model_file="unsloth.Q4_K_M.gguf", model_type="mistral", hf=True)
9
- tokenizer = AutoTokenizer.from_pretrained( "Detsutut/Igea-1B-qa")
10
 
11
 
12
  gen_pipeline = pipeline(
 
5
  import re
6
 
7
  # Initialize the model
8
+ model = AutoModelForCausalLM.from_pretrained("bmi-labmedinfo/Igea-1B-QA-v0.1-GGUF", model_file="unsloth.Q4_K_M.gguf", model_type="mistral", hf=True)
9
+ tokenizer = AutoTokenizer.from_pretrained( "bmi-labmedinfo/Igea-1B-QA-v0.1")
10
 
11
 
12
  gen_pipeline = pipeline(