frankjosh commited on
Commit
60d2710
·
verified ·
1 Parent(s): 877a41e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -24,7 +24,7 @@ if 'feedback' not in st.session_state:
24
  st.session_state.feedback = {}
25
 
26
  # Define subset size
27
- SUBSET_SIZE = 500 # Starting with 500 items for quick testing
28
 
29
  class TextDataset(Dataset):
30
  def __init__(self, texts: List[str], tokenizer, max_length: int = 512):
@@ -339,7 +339,6 @@ st.markdown("---")
339
  st.markdown(
340
  """
341
  Made with 🤖 using CodeT5 and Streamlit |
342
- GPU Status: {'🟢 Enabled' if torch.cuda.is_available() else '🔴 Disabled'} |
343
- Model: CodeT5-Small
344
  """
345
  )
 
24
  st.session_state.feedback = {}
25
 
26
  # Define subset size
27
+ SUBSET_SIZE = 5000 # Starting with 500 items for quick testing
28
 
29
  class TextDataset(Dataset):
30
  def __init__(self, texts: List[str], tokenizer, max_length: int = 512):
 
339
  st.markdown(
340
  """
341
  Made with 🤖 using CodeT5 and Streamlit |
342
+
 
343
  """
344
  )