Update app.py

app.py

@@ -1,9 +1,11 @@
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import streamlit as st
-model_id = "RWKV/rwkv-raven-1b5"
-model = AutoModelForCausalLM.from_pretrained(model_id)
-tokenizer = AutoTokenizer.from_pretrained(model_id)
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("samwit/koala-7b")
+
+model = AutoModelForCausalLM.from_pretrained("samwit/koala-7b")
 
 st.title("Raven Text Generator")
 st.write("Ask a question about ravens and get a response!")