alexkueck commited on
Commit
3d9cdb5
·
verified ·
1 Parent(s): 166cf16

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -22,6 +22,7 @@ import tempfile
22
 
23
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoModelForCausalLM, GPTNeoForCausalLM, GPT2Tokenizer, DistilBertTokenizer, DistilBertForQuestionAnswering
24
  from sentence_transformers import SentenceTransformer, util
 
25
  from typing import List, Dict
26
 
27
  import gradio as gr
 
22
 
23
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoModelForCausalLM, GPTNeoForCausalLM, GPT2Tokenizer, DistilBertTokenizer, DistilBertForQuestionAnswering
24
  from sentence_transformers import SentenceTransformer, util
25
+ from huggingface_hub import HfApi
26
  from typing import List, Dict
27
 
28
  import gradio as gr