Spaces:

limitedonly41
/

website_classification

Sleeping

limitedonly41 commited on 12 days ago

Commit

7a45750

•

1 Parent(s): dd5374d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,8 +16,8 @@ max_seq_length = 2048
 dtype = None  # Auto detection of dtype
 load_in_4bit = True  # Use 4-bit quantization to reduce memory usage
-peft_model_name = "limitedonly41/website_qwen2_7b_2"
 # Initialize model and tokenizer variables
 model = None
 tokenizer = None

 dtype = None  # Auto detection of dtype
 load_in_4bit = True  # Use 4-bit quantization to reduce memory usage
+# peft_model_name = "limitedonly41/website_qwen2_7b_2"
+peft_model_name = "limitedonly41/website_mistral7b_v02"
 # Initialize model and tokenizer variables
 model = None
 tokenizer = None