Hugging Face's logo

Spaces:
rodrigomasini
/
rephrase
Paused

App Files Community
rodrigomasini commited on
Commit
0e61e5c
·
1 Parent(s): 5034c15

Update app_v4.py

Browse files
Files changed (1) hide show
  1. app_v4.py +2 -1
app_v4.py CHANGED
@@ -1,4 +1,5 @@
1
  import torch
2
  import streamlit as st
3
 
4
- st.markdown(f"torch.cuda.memory_summary()")
 
 
1
  import torch
2
  import streamlit as st
3
 
4
+ viz = torch.cuda.memory_summary()
5
+ st.write(viz)