Merge pull request #14 from Sibbear1980/branch_kate

Update files (PEP8)

main.py

@@ -7,7 +7,7 @@
 app = FastAPI()
 tokenizer = BertTokenizerFast.from_pretrained(
     "blanchefort/rubert-base-cased-sentiment"
-)
+    )
 model = AutoModelForSequenceClassification.from_pretrained(
     "blanchefort/rubert-base-cased-sentiment", return_dict=True
 )
@@ -17,7 +17,7 @@
 @app.get("/")
 def root() -> dict:
     """
-    This function works from root 
+    This function works from root
     and demonstrates welcome message.
 
     Returns:
@@ -30,13 +30,12 @@ def root() -> dict:
 # функция для расчета модели,
 # в "input" передаются текст
 def predict(input: dict) -> dict:
-
     """
-    This function predicts 
+    This function predicts
 
     Args:
         input (dict): dict with text for prediction
-    
+
     Returns:
         dict: dict with the prediction of text sentiment
     """

test_main.py

@@ -45,7 +45,9 @@ def test_predict_russian_negative():
 
 # тест 6
 def test_predict_russian_neutral():
-    response = client.post("/predict/", json={"text": "Сегодня наступила весна"})
+    response = client.post("/predict/",
+                           json={"text": "Сегодня наступила весна"}
+                           )
     json_data = response.json()
     assert response.status_code == 200
     assert json_data["Результат:"] == "нейтральный"