Spaces:

frankjosh
/

repo_recommender

Sleeping

frankjosh commited on 22 days ago

Commit

c9bea94

verified ·

1 Parent(s): e688e26

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ if 'feedback' not in st.session_state:
     st.session_state.feedback = {}
 # Define subset size
-SUBSET_SIZE = 1000  # Starting with 500 items for quick testing
 class TextDataset(Dataset):
     def __init__(self, texts: List[str], tokenizer, max_length: int = 512):

     st.session_state.feedback = {}
 # Define subset size
+SUBSET_SIZE = 500  # Starting with 500 items for quick testing
 class TextDataset(Dataset):
     def __init__(self, texts: List[str], tokenizer, max_length: int = 512):