Detsutut commited on
Commit
6d520fd
1 Parent(s): a9a7a15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,8 +33,8 @@ def save_json(last_state: dict, pos_or_neg: str) -> None:
33
  ### /FEEDBACKS
34
 
35
  # Initialize the model
36
- model = AutoModelForCausalLM.from_pretrained("Detsutut/Igea-1B-instruct-v0.3-test4epochs-GGUF", model_file="unsloth.Q4_K_M.gguf", model_type="mistral", hf=True)
37
- tokenizer = AutoTokenizer.from_pretrained( "Detsutut/Igea-1B-instruct-v0.3-test4epochs")
38
 
39
  gen_pipeline = pipeline(
40
  "text-generation",
 
33
  ### /FEEDBACKS
34
 
35
  # Initialize the model
36
+ model = AutoModelForCausalLM.from_pretrained("bmi-labmedinfo/Igea-1B-instruct-GGUF", model_file="unsloth.Q4_K_M.gguf", model_type="mistral", hf=True)
37
+ tokenizer = AutoTokenizer.from_pretrained( "bmi-labmedinfo/Igea-1B-instruct")
38
 
39
  gen_pipeline = pipeline(
40
  "text-generation",