Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -44,14 +44,6 @@ geolocator = Nominatim(user_agent="skin-dashboard", timeout = 10)
|
|
44 |
# --- Load Model & Feature Extractor ---
|
45 |
@st.cache_resource
|
46 |
def load_image_model(token: str):
|
47 |
-
extractor = AutoFeatureExtractor.from_pretrained(
|
48 |
-
MODEL_NAME,
|
49 |
-
use_auth_token=token
|
50 |
-
)
|
51 |
-
model = AutoModelForImageClassification.from_pretrained(
|
52 |
-
MODEL_NAME,
|
53 |
-
use_auth_token=token
|
54 |
-
)
|
55 |
return pipeline(
|
56 |
"image-classification",
|
57 |
model=model,
|
@@ -61,16 +53,6 @@ def load_image_model(token: str):
|
|
61 |
|
62 |
@st.cache_resource
|
63 |
def load_llm(token: str):
|
64 |
-
|
65 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
66 |
-
LLM_NAME,
|
67 |
-
use_auth_token=token
|
68 |
-
)
|
69 |
-
# Use Seq2SeqLM for T5-style (text2text) models:
|
70 |
-
model = AutoModelForSeq2SeqLM.from_pretrained(
|
71 |
-
LLM_NAME,
|
72 |
-
use_auth_token=token,
|
73 |
-
)
|
74 |
return pipeline(
|
75 |
"text2text-generation",
|
76 |
model=model,
|
|
|
44 |
# --- Load Model & Feature Extractor ---
|
45 |
@st.cache_resource
|
46 |
def load_image_model(token: str):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
47 |
return pipeline(
|
48 |
"image-classification",
|
49 |
model=model,
|
|
|
53 |
|
54 |
@st.cache_resource
|
55 |
def load_llm(token: str):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
56 |
return pipeline(
|
57 |
"text2text-generation",
|
58 |
model=model,
|