Merge pull request #4 from kavlab/zvereva_ev

requirements.txt

@@ -6,3 +6,5 @@ transformers==4.35.0
 sentencepiece==0.1.99
 sacremoses==0.1.1
 translate==3.6.1
+fastapi==0.104.1
+uvicorn=0.24.0.

zvereva_ev/develop_api_app.py

@@ -0,0 +1,23 @@
+from fastapi import FastAPI
+from transformers import pipeline
+
+# Создание объекта FastAPI
+app = FastAPI()
+# Создание классификатора из библиотеки Hugging Face на основе пайплайна с типом "image-to-text"
+classifier = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
+
+
+@app.get("/")
+def root():
+    """
+    Приветственное сообщение
+    """
+    return {"message": "Follow the link http://127.0.0.1:8000/predict/ and get a description of the image"}
+
+
+@app.get("/predict/")
+async def predict():
+    """
+    Позволяет передать изображение 'image_result.jpg' для получения его описания
+    """
+    return classifier("image_result.jpg")[0]