Update app.py

app.py

@@ -12,6 +12,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import os
 import uuid
+import optimum
 
 
 welcome_message = """