|
@@ -15,4 +15,6 @@ def predict(inp):
|
|
|
return {labels[i]: float(prediction[i]) for i in range(1000)}
|
|
|
inputs = gr.inputs.Image()
|
|
|
outputs = gr.outputs.Label(num_top_classes=3)
|
|
|
-gr.Interface(fn=predict, inputs=inputs, outputs=outputs).launch()
|
|
|
+
|
|
|
+if __name__=='__main__':
|
|
|
+ gr.Interface(fn=predict, inputs=inputs, outputs=outputs).launch()
|