chrisjay commited on
Commit
d9cf4b7
β€’
1 Parent(s): ee1d26b

added link to model

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -3,7 +3,7 @@ from typing import Iterator
3
  import torch
4
  from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
5
 
6
- model_id = 'tosin/dialogpt_afriwoz_pidgin'
7
 
8
  #if torch.cuda.is_available():
9
  config = AutoConfig.from_pretrained(model_id)
 
3
  import torch
4
  from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
5
 
6
+ model_id = 'tosin/dialogpt_afriwoz_pidgin' # https://huggingface.co/tosin/dialogpt_afriwoz_pidgin?text=How+I+fit+chop+for+here%3F
7
 
8
  #if torch.cuda.is_available():
9
  config = AutoConfig.from_pretrained(model_id)