Update README.md

README.md

@@ -49,19 +49,16 @@ All you need to do is to run this in the terminal: <code>pip install gradio</cod
 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>
-  <pre>
-  def predict(inp):
-
-    inp = Image.fromarray(inp.astype('uint8'), 'RGB')
+def predict(inp):
+  inp = Image.fromarray(inp.astype('uint8'), 'RGB')
   
-    inp = transforms.ToTensor()(inp).unsqueeze(0)
+  inp = transforms.ToTensor()(inp).unsqueeze(0)
   
-    with torch.no_grad():
+  with torch.no_grad():
   
-      prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
+    prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
   
-    return {labels[i]: float(prediction[i]) for i in range(1000)}
-    </pre>
+  return {labels[i]: float(prediction[i]) for i in range(1000)}
 </code>
 </pre>
 </p>