Spaces:
Runtime error
Runtime error
jharrison27
commited on
Commit
•
d7135ca
1
Parent(s):
5eabe95
update word
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import numpy as np
|
|
5 |
|
6 |
# Mock data
|
7 |
mock_words = [
|
8 |
-
"apple", "banana", "
|
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.
|
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():
|