Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,14 +2,14 @@ from fastapi import FastAPI
|
|
2 |
import src.paraphrase.Paraphrase as Paraphrase
|
3 |
import src.translate.Translate as Translate
|
4 |
import nltk
|
5 |
-
nltk.download()
|
6 |
|
7 |
app = FastAPI(docs_url="/")
|
8 |
MTMODELS = {'enro': 'BlackKakapo/opus-mt-en-ro', 'roen': 'BlackKakapo/opus-mt-ro-en'}
|
9 |
|
10 |
@app.get("/")
|
11 |
def index():
|
12 |
-
return {
|
13 |
|
14 |
@app.get("/paraphrase")
|
15 |
def paraphrase(text: str, model: str):
|
|
|
2 |
import src.paraphrase.Paraphrase as Paraphrase
|
3 |
import src.translate.Translate as Translate
|
4 |
import nltk
|
5 |
+
nltk.download('punkt')
|
6 |
|
7 |
app = FastAPI(docs_url="/")
|
8 |
MTMODELS = {'enro': 'BlackKakapo/opus-mt-en-ro', 'roen': 'BlackKakapo/opus-mt-ro-en'}
|
9 |
|
10 |
@app.get("/")
|
11 |
def index():
|
12 |
+
return {'endpoints': ['/paraphrase', '/translate'], 'mtmodels': MTMODELS}
|
13 |
|
14 |
@app.get("/paraphrase")
|
15 |
def paraphrase(text: str, model: str):
|