Update app.py

app.py

@@ -25,9 +25,9 @@ model = torch.load("best_model-epoch=01-val_loss=3.00.ckpt")
 def preprocess_input(input):
  """Preprocess the input image for the PyTorch image classification model.
  Args:
-  input: A PIL Image object.
+ input: A PIL Image object.
  Returns:
-  A PyTorch tensor representing the preprocessed image.
+ A PyTorch tensor representing the preprocessed image.
  """
 
  # Resize the image to the expected size.
@@ -46,9 +46,9 @@ def preprocess_input(input):
 async def predict_endpoint(input: Any):
  """Predict the output of the PyTorch image classification model.
  Args:
-  input: A file containing the input image.
+ input: A file containing the input image.
  Returns:
-  A JSON object containing the prediction.
+ A JSON object containing the prediction.
  """
 
  # Load the image.
@@ -69,4 +69,4 @@ async def predict_endpoint(input: Any):
 
 if _name_ == "_main_":
  import uvicorn
- uvicorn.run(app, host="0.0.0.0", port=8000)
+ uvicorn.run(app, host="0.0.0.0", port=8000)