Update app.py

app.py

@@ -28,9 +28,14 @@ class preProcess:
         headers = [[] for i in range(0, len(df.columns))]
         for i in range(len(df.columns)):
             headers[i] = list(df[df.columns[i]])
+        
         zipped = list(zip(*headers))
         res = [' '.join(map(str,tups)) for tups in zipped]
-        input_format = ' labels ' + ' - '.join(list(df.columns)) +  ' values '  + ' , '.join(res)
+        if len(df.columns) < 2:
+          input_format = ' x-y values ' + ' - '.join(list(df.columns)) +  ' values '  + ' , '.join(res)
+
+        else:
+          input_format = ' labels ' + ' - '.join(list(df.columns)) +  ' values '  + ' , '.join(res)
 
         return input_format