feat: AMD compatibility

Author: LucaTedeschini

main.py

@@ -38,7 +38,9 @@
 disable_progress_bars()
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "true"
-device = "cuda" if torch.cuda.is_available() else "cpu"
+#AMD FIX
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model_name = 'pysentimiento/robertuito-base-cased'
 model_name = "Twitter/twhin-bert-base"
 model_name = "nickprock/setfit-italian-hate-speech"

requirements.txt

@@ -10,4 +10,7 @@ evaluate
 seaborn
 matplotlib
 google-generativeai
-load-dotenv
+rich
+questionary
+pysentimiento
+