jharrison27 commited on
Commit
d7135ca
1 Parent(s): 5eabe95

update word

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
 
6
  # Mock data
7
  mock_words = [
8
- "apple", "banana", "cherry", "date", # Fruits
9
  "car", "truck", "bus", "bicycle", # Vehicles
10
  "red", "blue", "green", "yellow", # Colors
11
  "cat", "dog", "rabbit", "hamster" # Pets
@@ -18,7 +18,7 @@ models = {
18
  'RoBERTa': 'roberta-base'
19
  }
20
 
21
- @st.cache(allow_output_mutation=True)
22
  def load_models():
23
  pipelines = {}
24
  for name, model_name in models.items():
 
5
 
6
  # Mock data
7
  mock_words = [
8
+ "apple", "banana", "grape", "date", # Fruits
9
  "car", "truck", "bus", "bicycle", # Vehicles
10
  "red", "blue", "green", "yellow", # Colors
11
  "cat", "dog", "rabbit", "hamster" # Pets
 
18
  'RoBERTa': 'roberta-base'
19
  }
20
 
21
+ @st.cache_resource
22
  def load_models():
23
  pipelines = {}
24
  for name, model_name in models.items():