Update app.py

app.py

@@ -4,15 +4,11 @@ import torch
 import numpy as np
 import gradio as gr
 
-if torch.cuda.is_available():
-    device = "cuda"
-else:
-    device = "cpu"
 
-model = RwkvForCausalLM.from_pretrained("StarRing2022/RWKV-430M-Pile-Alpaca",device_map='auto') #仅500MB,自训练，使用alpaca
+model = RwkvForCausalLM.from_pretrained("StarRing2022/RWKV-430M-Pile-Alpaca") 
 
-tokenizer = GPTNeoXTokenizerFast.from_pretrained("StarRing2022/RWKV-430M-Pile-Alpaca", add_special_tokens=True)
 
+tokenizer = GPTNeoXTokenizerFast.from_pretrained("StarRing2022/RWKV-430M-Pile-Alpaca", add_special_tokens=True)
 
 
 #rwkv with alpaca