Hugging Face's logo

Spaces:
pszemraj
/
small-instruct-streaming
Sleeping

App Files Community
pszemraj commited on
Commit
bf2a5db
·
verified ·
1 Parent(s): 1e1e796

update model to pszemraj/nanoT5-mid-2k-instruct

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import torch
11
  import gradio as gr
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
13
 
14
- model_id = "pszemraj/bart-large-mnli-dolly_hhrlhf-v1"
15
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
16
  logging.info(f"Running on device:\t {torch_device}")
17
  logging.info(f"CPU threads:\t {torch.get_num_threads()}")
 
11
  import gradio as gr
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
13
 
14
+ model_id = "pszemraj/nanoT5-mid-2k-instruct"
15
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
16
  logging.info(f"Running on device:\t {torch_device}")
17
  logging.info(f"CPU threads:\t {torch.get_num_threads()}")