use cpu
Browse files
app.py
CHANGED
@@ -21,9 +21,6 @@ warnings.filterwarnings('ignore')
|
|
21 |
# Load environment variables from .env file
|
22 |
load_dotenv()
|
23 |
|
24 |
-
# os.environ["TRANSFORMERS_CACHE"] = "./transformers_cache"
|
25 |
-
# os.makedirs("./transformers_cache", exist_ok=True)
|
26 |
-
|
27 |
class PDBSearchAssistant:
|
28 |
def __init__(self, model_name="google/flan-t5-large"):
|
29 |
# Set up HuggingFace pipeline with better model
|
|
|
21 |
# Load environment variables from .env file
|
22 |
load_dotenv()
|
23 |
|
|
|
|
|
|
|
24 |
class PDBSearchAssistant:
|
25 |
def __init__(self, model_name="google/flan-t5-large"):
|
26 |
# Set up HuggingFace pipeline with better model
|