Update main.py
Browse files
main.py
CHANGED
@@ -37,4 +37,5 @@ async def root():
|
|
37 |
@app.post("/predict")
|
38 |
async def root(text):
|
39 |
#return {"message": "Hello World"}
|
40 |
-
return generator(text,max_length=100, num_return_sequences=1)
|
|
|
|
37 |
@app.post("/predict")
|
38 |
async def root(text):
|
39 |
#return {"message": "Hello World"}
|
40 |
+
#return generator(text,max_length=100, num_return_sequences=1)
|
41 |
+
return result = pipeline.run(query=text)
|