Update app.py

app.py

@@ -31,7 +31,7 @@ from transformers import pipeline
 pipe = pipeline("text-generation",
                 model=model,
                 tokenizer= tokenizer,
-                torch_dtype=torch.bfloat32,
+                torch_dtype=torch.float32,
                 device_map="auto",
                 max_new_tokens = 512,
                 do_sample=True,