Spaces:
Running
Running
add space
Browse files
README.md
CHANGED
@@ -53,7 +53,8 @@ All you need to do is to run this in the terminal: <code>pip install gradio</cod
|
|
53 |
<p class="lg:col-span-2">
|
54 |
Here’s we define our image classification model prediction function in PyTorch (any framework, like TensorFlow, scikit-learn, JAX, or a plain Python will work as well):
|
55 |
<pre>
|
56 |
-
<code>
|
|
|
57 |
|
58 |
inp = Image.fromarray(inp.astype('uint8'), 'RGB')
|
59 |
|
|
|
53 |
<p class="lg:col-span-2">
|
54 |
Here’s we define our image classification model prediction function in PyTorch (any framework, like TensorFlow, scikit-learn, JAX, or a plain Python will work as well):
|
55 |
<pre>
|
56 |
+
<code>
|
57 |
+
def predict(inp):
|
58 |
|
59 |
inp = Image.fromarray(inp.astype('uint8'), 'RGB')
|
60 |
|