added ram usage

app.py

@@ -4,6 +4,7 @@ import streamlit as st
 import backend.aragpt
 import backend.home
 import backend.processor
+from backend.utils import get_current_ram_usage
 
 st.set_page_config(
     page_title="TEST", page_icon="📖", initial_sidebar_state="expanded", layout="wide"
@@ -34,3 +35,6 @@ st.sidebar.write(
 st.sidebar.write(
     "App source code available on [GitHub](https://github.com/WissamAntoun/Arabic-NLP-app)"
 )
+if st.sidebar.checkbox("Show RAM usage"):
+    ram = get_current_ram_usage()
+    st.sidebar.write("Ram usage: {:.2f}/{:.2f} GB".format(ram[0], ram[1]))

backend/services.py

@@ -1,12 +1,10 @@
 import json
 import os
-
 import requests
 from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline, set_seed
 from .modeling_gpt2 import GPT2LMHeadModel as GROVERLMHeadModel
 from .preprocess import ArabertPreprocessor
 
-
 # Taken and Modified from https://huggingface.co/spaces/flax-community/chef-transformer/blob/main/app.py
 class TextGeneration:
     def __init__(self):

backend/utils.py

@@ -0,0 +1,6 @@
+import psutil
+
+
+def get_current_ram_usage():
+    ram = psutil.virtual_memory()
+    return ram.available / 1024 / 1024 / 1024, ram.total / 1024 / 1024 / 1024

requirements.txt

@@ -5,5 +5,6 @@ PyArabic
 farasapy==0.0.14
 emoji==1.4.2
 awesome_streamlit
-torch
+torch==1.9.0
 transformers==4.10.0
+psutil==5.8.0