add pre element

README.md

@@ -36,6 +36,7 @@ All you need to do is to run this in the terminal: <code>pip install gradio</cod
 </h3>
 <p class="lg:col-span-2">
 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):
+<pre>
 <code>def predict(inp):
 
   inp = Image.fromarray(inp.astype('uint8'), 'RGB')
@@ -48,6 +49,7 @@ Here’s we define our image classification model prediction function in PyTorch
 
   return {labels[i]: float(prediction[i]) for i in range(1000)}
 </code>
+</pre>
 </p>
 
 <p class="lg:col-span-2">