Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,8 +3,8 @@ import streamlit as st
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-MODEL_PATH = "NetherQuartz/tatoeba-tok-multi-gemma-2-2b-merged-int4"
-DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+MODEL_PATH = "NetherQuartz/tatoeba-tok-multi-gemma-2-2b-merged"
+DEVICE = "mps" if torch.mps.is_available() else "cuda" if torch.cuda.is_available() else "cpu"
 
 st.set_page_config(
     page_icon="💬",