bug fix

app.py

@@ -19,14 +19,16 @@ def compute(dim):
     
     my_bar = st.progress(0)
     
-    for idx in api.train(iters=10000):
-        my_bar.progress(idx[0]/10000)
-    
+    for idx in api.train(iters=iters):
+        my_bar.progress(idx[0]/iters)
+    my_bar.progress(100)
     samples = np.array(api.model.sample(
             torch.tensor(api.scaled).float()).detach())
+    
+    
 
         # fig, ax = plt.subplots()
-    g = sns.jointplot(x=samples[:, 0], y=samples[:, 1], kind='kde',cmap=sns.color_palette("Blues", as_cmap=True),fill=True,label='Gaussian KDE',levels=50)
+    g = sns.jointplot(x=samples[:, 0], y=samples[:, 1], kind='kde',cmap=sns.color_palette("Blues", as_cmap=True),fill=True,label='Gaussian KDE',levels=1000)
 
     w = sns.scatterplot(x=api.scaled[:,0],y=api.scaled[:,1],ax=g.ax_joint,c='orange',marker='+',s=100,label='Real')
     st.pyplot(w.get_figure())
@@ -39,9 +41,8 @@ def compute(dim):
     
     return api.scaler.inverse_transform(samples)
     
-   
     
 if uploaded_file is not None:
-    dim = pd.read_csv(uploaded_file).shape[-1]
-    samples=compute(dim)
+    dims = len(uploaded_file.getvalue().decode("utf-8").split('\n')[0].split(','))-1
+    samples=compute(dims)
     st.download_button('Download generated CSV', pd.DataFrame(samples).to_csv(), 'text/csv')