Update app.py

app.py

@@ -3,6 +3,7 @@
 #!/usr/bin/env python
 # encoding: utf-8
 import torch
+torch.randn(10).cuda()
 import argparse
 from transformers import AutoModel, AutoTokenizer
 import gradio as gr