Detsutut commited on
Commit
6bb01ff
1 Parent(s): b258f29

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("bmi-labmedinfo/Igea-1B-instruct-v0.1-GGUF", model_file="unsloth.Q4_K_M.gguf", model_type="mistral", hf=True)
9
- tokenizer = AutoTokenizer.from_pretrained( "bmi-labmedinfo/Igea-1B-instruct-v0.1")
10
 
11
 
12
  gen_pipeline = pipeline(
 
5
  import re
6
 
7
  # Initialize the model
8
+ model = AutoModelForCausalLM.from_pretrained("Detsutut/Igea-1B-instruct-v0.2-GGUF", model_file="unsloth.Q4_K_M.gguf", model_type="mistral", hf=True)
9
+ tokenizer = AutoTokenizer.from_pretrained( "Detsutut/Igea-1B-instruct-v0.2")
10
 
11
 
12
  gen_pipeline = pipeline(