Update app.py

app.py

@@ -42,6 +42,7 @@ from inference import (
     load_model,
 )
 from app_modules.conversation import SeparatorStyle
+import torch
 torch.jit.script = lambda f: f
 
 def load_models():