Unggi commited on
Commit
60c3989
·
1 Parent(s): a9f32fb
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,8 +1,11 @@
1
  import gradio as gr
2
-
3
- import torch
4
  import transformers
5
 
 
 
 
 
 
6
 
7
 
8
  # saved_model
 
1
  import gradio as gr
 
 
2
  import transformers
3
 
4
+ # make function using import pip to install torch
5
+ import pip
6
+ pip.main(['install', 'torch'])
7
+
8
+ import torch
9
 
10
 
11
  # saved_model