Mizukiluke commited on
Commit
9d546e1
·
verified ·
1 Parent(s): 9873d77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@
3
  #!/usr/bin/env python
4
  # encoding: utf-8
5
  import torch
 
6
  import argparse
7
  from transformers import AutoModel, AutoTokenizer
8
  import gradio as gr
 
3
  #!/usr/bin/env python
4
  # encoding: utf-8
5
  import torch
6
+ torch.randn(10).cuda()
7
  import argparse
8
  from transformers import AutoModel, AutoTokenizer
9
  import gradio as gr