zhiqiulin commited on
Commit
6c3eb30
·
verified ·
1 Parent(s): 1e8e855

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -12,6 +12,8 @@ print(list_all_vqascore_models())
12
  model_pipe = None
13
 
14
  def initialize_model(model_name):
 
 
15
  print("Initializing model...")
16
  global model_pipe
17
  if model_pipe is None:
 
12
  model_pipe = None
13
 
14
  def initialize_model(model_name):
15
+ import torch
16
+ torch.jit.script = lambda f: f
17
  print("Initializing model...")
18
  global model_pipe
19
  if model_pipe is None: