hel8er commited on
Commit
0a51239
·
1 Parent(s): 186b04b
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
  from pydantic import BaseModel
4
 
5
  class TextQuery(BaseModel):
6
- text: str
7
 
8
 
9
  def predict(text: str):
 
3
  from pydantic import BaseModel
4
 
5
  class TextQuery(BaseModel):
6
+ inputs: str
7
 
8
 
9
  def predict(text: str):