sriirohit3107 commited on
Commit ·
d39ffe5
1
Parent(s): 5ecc51d
Updated app.py
Browse files
app.py
CHANGED
|
@@ -82,8 +82,8 @@ class LocalWebExplorerAgent:
|
|
| 82 |
self.device_str = "cpu" if use_cpu else ("cuda" if torch.cuda.is_available() else "cpu")
|
| 83 |
|
| 84 |
# Configure Entrez from environment variables if present
|
| 85 |
-
Entrez.email = os.getenv("ENTREZ_EMAIL"
|
| 86 |
-
Entrez.api_key = os.getenv("ENTREZ_API_KEY"
|
| 87 |
|
| 88 |
comps = _get_hf_components(self.device_str)
|
| 89 |
self.tokenizer = comps["tokenizer"]
|
|
|
|
| 82 |
self.device_str = "cpu" if use_cpu else ("cuda" if torch.cuda.is_available() else "cpu")
|
| 83 |
|
| 84 |
# Configure Entrez from environment variables if present
|
| 85 |
+
Entrez.email = os.getenv("ENTREZ_EMAIL")
|
| 86 |
+
Entrez.api_key = os.getenv("ENTREZ_API_KEY")
|
| 87 |
|
| 88 |
comps = _get_hf_components(self.device_str)
|
| 89 |
self.tokenizer = comps["tokenizer"]
|