AhmadT198 commited on
Commit
435ae7b
·
verified ·
1 Parent(s): b47601b

Update GPU acquiring duration

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