dcorcoran commited on
Commit
1d6967a
·
1 Parent(s): 1c99091

Added gradio to the predict endpoint result

Browse files
Files changed (2) hide show
  1. app/main.py +10 -1
  2. 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