Update app.py

app.py

@@ -14,6 +14,9 @@ state_dict = torch.load('up500Model.pt', map_location='cpu')
 model.load_state_dict(state_dict)
 model.eval()
 
+title = "VW Up! or Fiat 500"
+description = "Demo for classification of automobiles. To use it, simply upload your image, or click one of the examples to load them."
+
 imgTransforms = transforms.Compose([
     transforms.Resize(256),
     transforms.CenterCrop(224),
@@ -27,5 +30,5 @@ def predict(inp):
         prediction = torch.nn.functional.softmax(model(inp)[0], dim=0) 
     return {LABELS[i]: float(prediction[i]) for i in range(2)}
 
-iface = gr.Interface(predict, inputs=gr.inputs.Image(), outputs="label")
+iface = gr.Interface(predict, inputs=gr.inputs.Image(), outputs="label", title=title, description=description)
 iface.launch()