MoritzMMuller commited on
Commit
9ed5f58
·
verified ·
1 Parent(s): f30db10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=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=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,