shivansarora commited on
Commit
9da7e1d
·
verified ·
1 Parent(s): ee8e1c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -1,6 +1,15 @@
1
  import gradio as gr
2
  import os
3
  import json
 
 
 
 
 
 
 
 
 
4
  from parlai.core.opt import Opt
5
  from parlai.zoo.blender.blender_3B import download
6
  from parlai.core.agents import Agent
 
1
  import gradio as gr
2
  import os
3
  import json
4
+ import torch
5
+
6
+ _original_load = torch.load
7
+
8
+ def patched_load(*args, **kwargs):
9
+ kwargs['weights_only'] = False
10
+ return _original_load(*args, **kwargs)
11
+
12
+ torch.load = patched_load
13
  from parlai.core.opt import Opt
14
  from parlai.zoo.blender.blender_3B import download
15
  from parlai.core.agents import Agent