Switch to hosted model on Hugging Face

app/app.py

@@ -5,8 +5,8 @@ import torch
 # Load model and tokenizer
 @st.cache_resource
 def load_model():
-    model = AutoModelForSeq2SeqLM.from_pretrained("model/codementor-flan")
-    tokenizer = AutoTokenizer.from_pretrained("model/codementor-flan")
+    model = AutoModelForSeq2SeqLM.from_pretrained("Tuathe/codementor-flan")
+    tokenizer = AutoTokenizer.from_pretrained("Tuathe/codementor-flan")
     return model, tokenizer
 
 model, tokenizer = load_model()