Update app.py

app.py

@@ -29,7 +29,7 @@ def generate_data(noise=0):
 
 # Train the model
 def train_model(epochs, learning_rate):
-    print(f"🔄 Training started with {epochs} epochs and learning rate {learning_rate}...")  # Debugging
+    print(f"🔄 Training started with {epochs} epochs and learning rate {learning_rate}...")
 
     model = SimpleNN(1, 4, 1)
     optimizer = optim.SGD(model.parameters(), lr=learning_rate)
@@ -48,7 +48,7 @@ def train_model(epochs, learning_rate):
         optimizer.step()
         losses.append(loss.item())
 
-    print("✅ Training complete. Plotting loss...")  # Debugging
+    print("✅ Training complete. Plotting loss...")
 
     # Plot loss graph
     plt.figure(figsize=(5, 3))
@@ -77,8 +77,9 @@ iface = gr.Interface(
     description="Train a simple neural network and visualize the loss curve.",
 )
 
-print("✅ Gradio is about to launch...")  # Debugging
+print("✅ Gradio is about to launch...")
 iface.launch()
 
 
 
+