Update app.py

app.py

@@ -5,8 +5,9 @@ import spaces
 
 model_name = "MBZUAI-Paris/Atlas-Chat-27B"
 model = AutoModelForCausalLM.from_pretrained(
+    "text-generation",
     model_name, 
-    load_in_4bit=True,
+     model_kwargs={"torch_dtype": torch.bfloat16},
     device_map="auto"
 )
 tokenizer = AutoTokenizer.from_pretrained(model_name)