Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
37 |
-
tokenizer = AutoTokenizer.from_pretrained( "
|
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",
|