Update app/main.py

app/main.py

@@ -31,10 +31,13 @@ def home():
 
 @app.post("/upload/")
 def upload_csv(file:UploadFile=File(...)):
-    global dataset
-    dataset=pd.read_csv(file.file)
-    file.file.close()
-    return {"filename": file.filename}
+    try:
+        global dataset
+        dataset=pd.read_csv(file.file)
+        file.file.close()
+        return {"filename": file.filename}
+    else:
+        return {"message":"Pleae upload a CSV file"}
 
 @app.post("/train/")
 def training():