TiberiuCristianLeon commited on
Commit
5f8f1a2
·
verified ·
1 Parent(s): 8fd2d1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -3,13 +3,12 @@ import src.Paraphrase as Paraphrase
3
  import src.Translate as Translate
4
  from typing import Optional
5
 
6
- app = FastAPI()
7
- # app = FastAPI(docs_url="/")
8
  MTMODELS = {'enro': 'BlackKakapo/opus-mt-en-ro',
9
  'roen': 'BlackKakapo/opus-mt-ro-en',
10
  'gemma': 'Gargaz/gemma-2b-romanian-better'}
11
 
12
- @app.get("/")
13
  def index():
14
  return {'endpoints': ['/paraphrase', '/translate'], 'mtmodels': MTMODELS}
15
 
 
3
  import src.Translate as Translate
4
  from typing import Optional
5
 
6
+ app = FastAPI(docs_url="/")
 
7
  MTMODELS = {'enro': 'BlackKakapo/opus-mt-en-ro',
8
  'roen': 'BlackKakapo/opus-mt-ro-en',
9
  'gemma': 'Gargaz/gemma-2b-romanian-better'}
10
 
11
+ @app.get("/endpoints")
12
  def index():
13
  return {'endpoints': ['/paraphrase', '/translate'], 'mtmodels': MTMODELS}
14