Added gradio to the predict endpoint result
Browse files- app/main.py +10 -1
- requirements.txt +2 -1
app/main.py
CHANGED
|
@@ -3,6 +3,7 @@ from fastapi.middleware.cors import CORSMiddleware
|
|
| 3 |
from contextlib import asynccontextmanager
|
| 4 |
from PIL import Image
|
| 5 |
import io
|
|
|
|
| 6 |
|
| 7 |
from app.services.embedding_service import EmbeddingService
|
| 8 |
from app.services.similarity_service import SimilarityService
|
|
@@ -78,4 +79,12 @@ async def predict(file: UploadFile = File(...)):
|
|
| 78 |
types=ocr_data.get("types"),
|
| 79 |
moves=ocr_data.get("moves"),
|
| 80 |
similar_cards=similar_cards
|
| 81 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
from contextlib import asynccontextmanager
|
| 4 |
from PIL import Image
|
| 5 |
import io
|
| 6 |
+
import gradio as gr
|
| 7 |
|
| 8 |
from app.services.embedding_service import EmbeddingService
|
| 9 |
from app.services.similarity_service import SimilarityService
|
|
|
|
| 79 |
types=ocr_data.get("types"),
|
| 80 |
moves=ocr_data.get("moves"),
|
| 81 |
similar_cards=similar_cards
|
| 82 |
+
)
|
| 83 |
+
|
| 84 |
+
gr.Interface(
|
| 85 |
+
fn=predict,
|
| 86 |
+
inputs=gr.File(label="Upload Pokémon Card"),
|
| 87 |
+
outputs=gr.JSON(),
|
| 88 |
+
title="Pokemon Card Image Processor",
|
| 89 |
+
description="Upload a Pokémon card image to extract its data and find similar cards."
|
| 90 |
+
).launch()
|
requirements.txt
CHANGED
|
@@ -10,4 +10,5 @@ opencv-python
|
|
| 10 |
pydantic
|
| 11 |
pydantic-settings
|
| 12 |
python-multipart
|
| 13 |
-
requests
|
|
|
|
|
|
| 10 |
pydantic
|
| 11 |
pydantic-settings
|
| 12 |
python-multipart
|
| 13 |
+
requests
|
| 14 |
+
gradio>=3.40
|