Update GPU acquiring duration
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ index_to_class = dict((v,k) for k,v in class_to_index.items())
|
|
48 |
names_to_ids = lambda labels: np.array([class_to_index.get(x) for x in labels])
|
49 |
train_labels = names_to_ids(labels)
|
50 |
|
51 |
-
@spaces.GPU
|
52 |
def load_model():
|
53 |
|
54 |
|
@@ -96,7 +96,7 @@ def load_model():
|
|
96 |
|
97 |
model = load_model()
|
98 |
|
99 |
-
@spaces.GPU
|
100 |
def predict(tweet):
|
101 |
print("ENTERED", tweet)
|
102 |
seq = get_sequences(tokenizer, [tweet])
|
|
|
48 |
names_to_ids = lambda labels: np.array([class_to_index.get(x) for x in labels])
|
49 |
train_labels = names_to_ids(labels)
|
50 |
|
51 |
+
@spaces.GPU(duration=200)
|
52 |
def load_model():
|
53 |
|
54 |
|
|
|
96 |
|
97 |
model = load_model()
|
98 |
|
99 |
+
@spaces.GPU(duration=100)
|
100 |
def predict(tweet):
|
101 |
print("ENTERED", tweet)
|
102 |
seq = get_sequences(tokenizer, [tweet])
|