Hugging Face's logo

Spaces:
bmi-labmedinfo
/
igea-pretrained
Runtime error

App Files Community
Detsutut commited on
Commit
9fb2fed
•
1 Parent(s): 106e81e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  import re
6
 
7
  # Initialize the model
8
- model = AutoModelForCausalLM.from_pretrained("bmi-labmedinfo/Igea-1B-v0.1-GGUF", model_file="igea-1b-v0.1-q4_k_m.gguf", model_type="mistral", hf=True)
9
  tokenizer = AutoTokenizer.from_pretrained( "bmi-labmedinfo/Igea-1B-v0.1")
10
 
11
 
 
5
  import re
6
 
7
  # Initialize the model
8
+ model = AutoModelForCausalLM.from_pretrained("bmi-labmedinfo/Igea-1B-v0.1-GGUF", model_file="igea-1b-v0.0.1-q4_k_m.gguf", model_type="mistral", hf=True)
9
  tokenizer = AutoTokenizer.from_pretrained( "bmi-labmedinfo/Igea-1B-v0.1")
10
 
11