Hugging Face's logo

Spaces:
GooooGLE
/
myFirstTestSpace
Sleeping

App Files Community
GooooGLE commited on
Commit
5270bf3
·
verified ·
1 Parent(s): bfd65de

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -6,7 +6,7 @@ from transformers import pipeline
6
 
7
  app = FastAPI()
8
 
9
- pipe_flan = pipeline("text2text-generation", model="GooooGLE/testVulModel")
10
 
11
  variable = os.getenv('test_variable')
12
  secret = os.getenv('test_secret')
 
6
 
7
  app = FastAPI()
8
 
9
+ pipe_flan = pipeline("text2text-generation", model="GooooGLE/testVulModel",ignore_mismatched_sizes=True)
10
 
11
  variable = os.getenv('test_variable')
12
  secret = os.getenv('test_secret')