Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import streamlit as st
 model_id = "RWKV/rwkv-raven-1b5"
-model = AutoModelForCausalLM.from_pretrained(model_id).to(0)
+model = AutoModelForCausalLM.from_pretrained(model_id)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 st.title("Raven Text Generator")
@@ -15,7 +15,7 @@ if st.button("Generate Response"):
     if question.strip() != "":
         # Generate response based on the provided question
         prompt = f"### Instruction: {question}\n### Response:"
-        inputs = tokenizer(prompt, return_tensors="pt").to(0)
+        inputs = tokenizer(prompt, return_tensors="pt")
         output = model.generate(inputs["input_ids"], max_new_tokens=100)
         generated_text = tokenizer.decode(output[0].tolist(), skip_special_tokens=True)
         st.markdown("## Generated Response")