Spaces:
Running
Running
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):
|