MoritzMMuller commited on
Commit
f30db10
·
verified ·
1 Parent(s): 22353f5

Update app.py

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