Removed secrets from history
Browse files- app.py +8 -1
- requirements.txt +2 -6
app.py
CHANGED
@@ -19,6 +19,8 @@ from eva_vit import create_eva_vit_g
|
|
19 |
import requests
|
20 |
from io import BytesIO
|
21 |
import os
|
|
|
|
|
22 |
|
23 |
token = os.getenv("HF_TOKEN")
|
24 |
if not token:
|
@@ -379,7 +381,12 @@ class SkinGPT4(nn.Module):
|
|
379 |
|
380 |
|
381 |
def load_model(model_path):
|
382 |
-
|
|
|
|
|
|
|
|
|
|
|
383 |
model.to(device)
|
384 |
model.eval()
|
385 |
return model
|
|
|
19 |
import requests
|
20 |
from io import BytesIO
|
21 |
import os
|
22 |
+
from huggingface_hub import hf_hub_download
|
23 |
+
|
24 |
|
25 |
token = os.getenv("HF_TOKEN")
|
26 |
if not token:
|
|
|
381 |
|
382 |
|
383 |
def load_model(model_path):
|
384 |
+
model_path = hf_hub_download(
|
385 |
+
repo_id="KeerthiVM/SkinCancerDiagnosis",
|
386 |
+
filename="dermnet_finetuned_version1.pth",
|
387 |
+
)
|
388 |
+
# model = SkinGPT4(vit_checkpoint_path="dermnet_finetuned_version1.pth")
|
389 |
+
model = SkinGPT4(vit_checkpoint_path=model_path)
|
390 |
model.to(device)
|
391 |
model.eval()
|
392 |
return model
|
requirements.txt
CHANGED
@@ -8,12 +8,8 @@ pillow
|
|
8 |
fpdf
|
9 |
nest_asyncio
|
10 |
sentencepiece
|
11 |
-
qdrant_client
|
12 |
-
langchain
|
13 |
-
langchain_community
|
14 |
scikit-learn
|
15 |
-
langchain_openai
|
16 |
nest_asyncio
|
17 |
sentence_transformers
|
18 |
-
|
19 |
-
|
|
|
8 |
fpdf
|
9 |
nest_asyncio
|
10 |
sentencepiece
|
|
|
|
|
|
|
11 |
scikit-learn
|
|
|
12 |
nest_asyncio
|
13 |
sentence_transformers
|
14 |
+
huggingface_hub
|
15 |
+
timm
|