Update main.py

main.py

@@ -2,7 +2,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import os
 
-os.environ['SENTENCE_TRANSFORMERS_HOME'] = './.cache'
 
 model = AutoModelForCausalLM.from_pretrained(
     "E-Hospital/open-orca-platypus-2-lora-medical",
@@ -19,8 +18,6 @@ def ask_bot(question):
   response = generated_text.split("->:")[-1]
   return response
 
-ask_bot("I have diabetes. What should I do?")
-
 import mysql.connector
 import re
 from datetime import datetime