JustKiddo commited on
Commit
a42b9fc
·
verified ·
1 Parent(s): ce90c8f

google-t5/t5-small

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -94,7 +94,8 @@ class ContentFilter:
94
  content_filter = ContentFilter()
95
 
96
  # Initialize Hugging Face client
97
- client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
98
 
99
  # Load dataset (optional)
100
  dataset = load_dataset("JustKiddo/KiddosVault")
 
94
  content_filter = ContentFilter()
95
 
96
  # Initialize Hugging Face client
97
+ #client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
98
+ client = InferenceClient("google-t5/t5-small")
99
 
100
  # Load dataset (optional)
101
  dataset = load_dataset("JustKiddo/KiddosVault")