Update app.py

app.py

@@ -9,7 +9,7 @@ description = """ try this space to build downstream applications with [CohereFo
 device = "cuda"
 checkpoint = "CohereForAI/aya-101"
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint, torch_dtype=torch.float16, low_cpu_mem_usage=True, device_map=device)
+model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint, torch_dtype=torch.bfloat16, low_cpu_mem_usage=True, device_map=device)
 
 @spaces.GPU
 def aya(text):