ssocean commited on
Commit
23551ad
·
verified ·
1 Parent(s): b7157af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ tokenizer = None
15
 
16
  @spaces.GPU(duration=60, enable_queue=True)
17
  def predict(title, abstract):
 
18
  if model is None:
19
  model = AutoModelForSequenceClassification.from_pretrained(
20
  model_path,
 
15
 
16
  @spaces.GPU(duration=60, enable_queue=True)
17
  def predict(title, abstract):
18
+ global model, tokenizer
19
  if model is None:
20
  model = AutoModelForSequenceClassification.from_pretrained(
21
  model_path,