jsvs commited on
Commit
d1d4338
·
1 Parent(s): 9b6493b

udpate app file with use_auth_token

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ app = FastAPI()
7
  # This function will be able to generate text
8
  # given an input.
9
  pipe = pipeline("text2text-generation",
10
- model="visoc/Text2Question")
11
 
12
  # Define a function to handle the GET request at `/generate`
13
  # The generate() function is defined as a FastAPI route that takes a
 
7
  # This function will be able to generate text
8
  # given an input.
9
  pipe = pipeline("text2text-generation",
10
+ model="visoc/Text2Question",use_auth_token=True)
11
 
12
  # Define a function to handle the GET request at `/generate`
13
  # The generate() function is defined as a FastAPI route that takes a