Update app.py

app.py

@@ -391,6 +391,8 @@
 
 # if __name__ == "__main__":
 #     main()
+
+
 import streamlit as st
 import matplotlib.pyplot as plt
 import torch