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