yakine commited on
Commit
dbb92e9
·
verified ·
1 Parent(s): 20981ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from pydantic import BaseModel
4
  import pandas as pd
5
  import os
6
  import requests
7
- from transformers import GPT2LMHeadModel, GPT2Tokenizer, AutoTokenizer
8
  from io import StringIO
9
  from fastapi.middleware.cors import CORSMiddleware
10
  from huggingface_hub import HfFolder
 
4
  import pandas as pd
5
  import os
6
  import requests
7
+ from transformers import GPT2LMHeadModel, GPT2Tokenizer, AutoTokenizer, pipeline
8
  from io import StringIO
9
  from fastapi.middleware.cors import CORSMiddleware
10
  from huggingface_hub import HfFolder