Update app.py

app.py

@@ -10,6 +10,8 @@ import traceback
 
 import json
 
+import torch
+
 import logging
 
 from flask import Flask