Update app.py
Browse files
app.py
CHANGED
|
@@ -113,7 +113,14 @@ def main():
|
|
| 113 |
device = config['device']
|
| 114 |
model = load_model(config, device=device)
|
| 115 |
# Define Gradio interface for inference
|
| 116 |
-
def
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 117 |
return run_inference(input_image, model)
|
| 118 |
# Create Gradio Tabs
|
| 119 |
with gr.Blocks() as demo:
|
|
|
|
| 113 |
device = config['device']
|
| 114 |
model = load_model(config, device=device)
|
| 115 |
# Define Gradio interface for inference
|
| 116 |
+
def quick_predict(input_image):
|
| 117 |
+
"""Quick and simple check.
|
| 118 |
+
|
| 119 |
+
Args:
|
| 120 |
+
input_image: Image to analyze
|
| 121 |
+
Returns:
|
| 122 |
+
result: String
|
| 123 |
+
"""
|
| 124 |
return run_inference(input_image, model)
|
| 125 |
# Create Gradio Tabs
|
| 126 |
with gr.Blocks() as demo:
|