habulaj commited on
Commit
7730d87
·
verified ·
1 Parent(s): 9d98d14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,5 +1,4 @@
1
  from fastapi import FastAPI, HTTPException
2
- from pydantic import BaseModel
3
  import yake
4
 
5
  app = FastAPI()
@@ -27,8 +26,8 @@ async def extract_keywords(text: str, max_ngram_size: int = 3, num_of_keywords:
27
  # Extraindo as palavras-chave
28
  keywords = custom_kw_extractor.extract_keywords(text)
29
 
30
- # Retornando apenas as palavras-chave, sem as pontuações
31
- keyword_list = [kw[1] for kw in keywords]
32
 
33
  return {"keywords": keyword_list}
34
 
 
1
  from fastapi import FastAPI, HTTPException
 
2
  import yake
3
 
4
  app = FastAPI()
 
26
  # Extraindo as palavras-chave
27
  keywords = custom_kw_extractor.extract_keywords(text)
28
 
29
+ # Extrai apenas as palavras-chave (segundo valor de cada tupla)
30
+ keyword_list = [kw[1] for kw in keywords] # Aqui pegamos o segundo valor de cada tupla (a palavra-chave)
31
 
32
  return {"keywords": keyword_list}
33