add new import

app.py

@@ -5,6 +5,7 @@ Small demo application to explore Gradio.
 import argparse
 import os
 from functools import partial
+import torch
 
 import gradio as gr
 from PIL import Image