Update app.py

app.py

@@ -7,7 +7,7 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",
     model=model,
-    torch_dtype=torch.float16,
+    torch_dtype=torch.bfloat16,
 )
 
 prompt = "What is 6534+2343?"