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:
|