Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,13 +37,13 @@ def index(request: Request):
|
|
| 37 |
# host_url = request.url.scheme + "://" + request.url.netloc
|
| 38 |
# return {"host_url": host_url, 'endpoints': ['/paraphrase', '/translate', f'{host_url}/docs', f'{host_url}/redoc', f'{host_url}/openapi.json'], 'mtmodels': MTMODELS}
|
| 39 |
|
| 40 |
-
@app.get("/paraphrase")
|
| 41 |
-
def paraphrase(text: str, model: str
|
| 42 |
resultValue, exception = Paraphrase.paraphraseParaphraseMethod(text, model)
|
| 43 |
return {"input": text, "translation": resultValue, "exception": exception}
|
| 44 |
|
| 45 |
-
@app.get("/translate")
|
| 46 |
-
def translate(text: str, model: Optional[str] = MTMODELS['gemma']
|
| 47 |
# resultValue, exception = Translate.paraphraseTranslateMethod(text, model)
|
| 48 |
resultValue: str = Translate.gemma_direct(text, model)
|
| 49 |
return {"input": text, "result": resultValue, "model": model}
|
|
|
|
| 37 |
# host_url = request.url.scheme + "://" + request.url.netloc
|
| 38 |
# return {"host_url": host_url, 'endpoints': ['/paraphrase', '/translate', f'{host_url}/docs', f'{host_url}/redoc', f'{host_url}/openapi.json'], 'mtmodels': MTMODELS}
|
| 39 |
|
| 40 |
+
@app.get("/paraphrase", operation_id="get_paraphrase", description="Paraphrase text", tags=["paraphrase"], summary="Paraphrase text")
|
| 41 |
+
def paraphrase(text: str, model: str):
|
| 42 |
resultValue, exception = Paraphrase.paraphraseParaphraseMethod(text, model)
|
| 43 |
return {"input": text, "translation": resultValue, "exception": exception}
|
| 44 |
|
| 45 |
+
@app.get("/translate", operation_id="get_translate", description="Translate text", tags=["translate"], summary="Translate text")
|
| 46 |
+
def translate(text: str, model: Optional[str] = MTMODELS['gemma']):
|
| 47 |
# resultValue, exception = Translate.paraphraseTranslateMethod(text, model)
|
| 48 |
resultValue: str = Translate.gemma_direct(text, model)
|
| 49 |
return {"input": text, "result": resultValue, "model": model}
|