FlameF0X commited on
Commit
d319da4
·
verified ·
1 Parent(s): c161fad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -233,10 +233,10 @@ def load_model():
233
  device = "cuda" if torch.cuda.is_available() else "cpu"
234
  print(f"Loading FWKV-ROSA from Hub on {device} ...")
235
  try:
236
- model = FWKVLanguageModel.from_pretrained("FlameF0X/FWKV-ROSA")
237
  model = model.to(device)
238
  model.eval()
239
- tokenizer = AutoTokenizer.from_pretrained("FlameF0X/FWKV-ROSA")
240
  status = "FWKV-ROSA chat model ready!"
241
  except Exception as e:
242
  model, tokenizer = None, None
@@ -388,7 +388,7 @@ def chat_function(message, history):
388
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
389
  gr.Markdown(f"""
390
  # ⚡ FWKV-ROSA Chat
391
- **Model:** [FlameF0X/FWKV-ROSA](https://huggingface.co/FlameF0X/FWKV-ROSA)
392
  *{load_status}*
393
 
394
  This is a 56M‑parameter recurrent LM trained with the RWKV‑8 ROSA
 
233
  device = "cuda" if torch.cuda.is_available() else "cpu"
234
  print(f"Loading FWKV-ROSA from Hub on {device} ...")
235
  try:
236
+ model = FWKVLanguageModel.from_pretrained("FWKV/FWKV-ROSA")
237
  model = model.to(device)
238
  model.eval()
239
+ tokenizer = AutoTokenizer.from_pretrained("FWKV/FWKV-ROSA")
240
  status = "FWKV-ROSA chat model ready!"
241
  except Exception as e:
242
  model, tokenizer = None, None
 
388
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
389
  gr.Markdown(f"""
390
  # ⚡ FWKV-ROSA Chat
391
+ **Model:** [FKWV/FWKV-ROSA](https://huggingface.co/FWKV/FWKV-ROSA)
392
  *{load_status}*
393
 
394
  This is a 56M‑parameter recurrent LM trained with the RWKV‑8 ROSA