Update app.py

app.py

@@ -1,9 +1,10 @@
 import streamlit as st
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoTokenizer, FalconModel
+import torch
 
 device = "cuda"
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
+tokenizer = AutoTokenizer.from_pretrained("Rocketknight1/falcon-rw-1b")
+model = FalconModel.from_pretrained("Rocketknight1/falcon-rw-1b")
 
 model.to(device)