Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -69,7 +69,7 @@ tokenizer.padding_side = 'left'
|
|
| 69 |
model_config = PeftConfig.from_pretrained("Tonic/GaiaMiniMed")
|
| 70 |
# Load the PEFT model with the specified configuration
|
| 71 |
peft_model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-7b-instruct", config=model_config)
|
| 72 |
-
peft_model = PeftModel.from_pretrained(
|
| 73 |
peft_model = PeftModel.from_pretrained(peft_model, "Tonic/GaiaMiniMed")
|
| 74 |
|
| 75 |
|
|
@@ -81,9 +81,9 @@ peft_model = PeftModel.from_pretrained(peft_model, "Tonic/GaiaMiniMed")
|
|
| 81 |
#peft_model = AutoModelForCausalLM.from_pretrained(base_model_id, config=model_config)
|
| 82 |
|
| 83 |
# Load the PEFT model
|
| 84 |
-
# peft_config = PeftConfig.from_pretrained("Tonic/mistralmed"
|
| 85 |
# peft_model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True)
|
| 86 |
-
# peft_model = PeftModel.from_pretrained(peft_model, "Tonic/mistralmed"
|
| 87 |
|
| 88 |
class ChatBot:
|
| 89 |
def __init__(self, system_prompt="You are an expert medical analyst:"):
|
|
|
|
| 69 |
model_config = PeftConfig.from_pretrained("Tonic/GaiaMiniMed")
|
| 70 |
# Load the PEFT model with the specified configuration
|
| 71 |
peft_model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-7b-instruct", config=model_config)
|
| 72 |
+
peft_model = PeftModel.from_pretrained(model="Tonic/GaiaMiniMed")
|
| 73 |
peft_model = PeftModel.from_pretrained(peft_model, "Tonic/GaiaMiniMed")
|
| 74 |
|
| 75 |
|
|
|
|
| 81 |
#peft_model = AutoModelForCausalLM.from_pretrained(base_model_id, config=model_config)
|
| 82 |
|
| 83 |
# Load the PEFT model
|
| 84 |
+
# peft_config = PeftConfig.from_pretrained("Tonic/mistralmed")
|
| 85 |
# peft_model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True)
|
| 86 |
+
# peft_model = PeftModel.from_pretrained(peft_model, "Tonic/mistralmed")
|
| 87 |
|
| 88 |
class ChatBot:
|
| 89 |
def __init__(self, system_prompt="You are an expert medical analyst:"):
|