Grandediw commited on
Commit
39e4b7b
·
verified ·
1 Parent(s): 17ce963

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def load_model():
16
 
17
  # Load the LoRA adapter weights
18
  # Replace "Grandediw/lora_model_finetuned" with your actual LoRA model repo
19
- model = PeftModel.from_pretrained(base_model, "jacopoda/lora_model")
20
 
21
  # Create a pipeline for text generation
22
  chat_pipeline = pipeline(
 
16
 
17
  # Load the LoRA adapter weights
18
  # Replace "Grandediw/lora_model_finetuned" with your actual LoRA model repo
19
+ model = PeftModel.from_pretrained(base_model, "Grandediw/lora_model_finetuned")
20
 
21
  # Create a pipeline for text generation
22
  chat_pipeline = pipeline(