Spaces:
Paused
Paused
Commit
•
ac44131
1
Parent(s):
5ae0d54
Update main.py
Browse files
main.py
CHANGED
|
@@ -2,7 +2,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 2 |
import torch
|
| 3 |
import os
|
| 4 |
|
| 5 |
-
os.environ['SENTENCE_TRANSFORMERS_HOME'] = './.cache'
|
| 6 |
|
| 7 |
model = AutoModelForCausalLM.from_pretrained(
|
| 8 |
"E-Hospital/open-orca-platypus-2-lora-medical",
|
|
@@ -19,8 +18,6 @@ def ask_bot(question):
|
|
| 19 |
response = generated_text.split("->:")[-1]
|
| 20 |
return response
|
| 21 |
|
| 22 |
-
ask_bot("I have diabetes. What should I do?")
|
| 23 |
-
|
| 24 |
import mysql.connector
|
| 25 |
import re
|
| 26 |
from datetime import datetime
|
|
|
|
| 2 |
import torch
|
| 3 |
import os
|
| 4 |
|
|
|
|
| 5 |
|
| 6 |
model = AutoModelForCausalLM.from_pretrained(
|
| 7 |
"E-Hospital/open-orca-platypus-2-lora-medical",
|
|
|
|
| 18 |
response = generated_text.split("->:")[-1]
|
| 19 |
return response
|
| 20 |
|
|
|
|
|
|
|
| 21 |
import mysql.connector
|
| 22 |
import re
|
| 23 |
from datetime import datetime
|