Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ geolocator = Nominatim(user_agent="skin-dashboard", timeout = 10)
|
|
46 |
def load_image_model(token: str):
|
47 |
return pipeline(
|
48 |
"image-classification",
|
49 |
-
model=
|
50 |
feature_extractor=extractor,
|
51 |
device=0 # set to GPU index or -1 for CPU
|
52 |
)
|
@@ -55,7 +55,7 @@ def load_image_model(token: str):
|
|
55 |
def load_llm(token: str):
|
56 |
return pipeline(
|
57 |
"text2text-generation",
|
58 |
-
model=
|
59 |
tokenizer=tokenizer,
|
60 |
device_map="auto", # or device=0 for single GPU / -1 for CPU
|
61 |
max_length=10000,
|
|
|
46 |
def load_image_model(token: str):
|
47 |
return pipeline(
|
48 |
"image-classification",
|
49 |
+
model=MODEL_NAME,
|
50 |
feature_extractor=extractor,
|
51 |
device=0 # set to GPU index or -1 for CPU
|
52 |
)
|
|
|
55 |
def load_llm(token: str):
|
56 |
return pipeline(
|
57 |
"text2text-generation",
|
58 |
+
model=LLM_NAME,
|
59 |
tokenizer=tokenizer,
|
60 |
device_map="auto", # or device=0 for single GPU / -1 for CPU
|
61 |
max_length=10000,
|