Spaces:

pszemraj
/

small-instruct-streaming

Sleeping

pszemraj commited on May 23, 2023

Commit

935c638

1 Parent(s): 88a5766

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ import torch
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
-model_id = "pszemraj/flan-t5-large-instruct-dolly_hhrlhf"
 torch_device = "cuda" if torch.cuda.is_available() else "cpu"
 logging.info(f"Running on device:\t {torch_device}")
 logging.info(f"CPU threads:\t {torch.get_num_threads()}")

 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
+model_id = "pszemraj/bart-large-mnli-dolly_hhrlhf-v1"
 torch_device = "cuda" if torch.cuda.is_available() else "cpu"
 logging.info(f"Running on device:\t {torch_device}")
 logging.info(f"CPU threads:\t {torch.get_num_threads()}")