Update app.py

Fix missing import

app.py

@@ -29,7 +29,9 @@ import gradio as gr
 
 import torch
 from transformers import TorchAoConfig, AutoModelForCausalLM, AutoTokenizer
+import torchao.quantization.pt2e as pt2e
 from torchao.quantization import Float8DynamicActivationFloat8WeightConfig, Float8WeightOnlyConfig
+
 # quant_config = Float8WeightOnlyConfig()
 quant_config = Float8DynamicActivationFloat8WeightConfig()
 quantization_config = TorchAoConfig(quant_type=quant_config)