kalilurrahman/tnelectionpredictor2026
0
1�
2 ���i� � �< � S SK Jr S SKJrJrJrJr " S S5 rg)� )�pipeline)�List�Dict�Any�Tuplec �Z � \ rS rSrS S\4S jjrS S\S\\ S\S\\\\ 4 4S jjr3S r g4)
�ElectionPredictor� �5model_namec � � Xl [ SU R S35 [ SU R U R SS9U l [ S5 g ! [ a( n[ SU R SU 35 S U l e S nAff = f) NzLoading prediction model: z...zzero-shot-classification�����)�model� tokenizer�devicez%Prediction model loaded successfully.z Error loading prediction model z: )r �printr �6classifier� Exception)�selfr �es �_C:\Users\HP\OneDrive\Documents\GitHub\tn-election-predictor\tn-predictor-final\src\predictor.py�__init__�ElectionPredictor.__init__ s� � �$��
�*�4�?�?�*;�3�?�@� � &�*��o�o� �/�/�� �D�O�
�9� :��� � �3�D�O�O�3D�B�q�c�J� K� "�D�O� �� �s �0A �7B� #A?�?B�8text_input�candidate_or_party_names�num_predictions�returnc � � U R ( d S/$ U( a UR 5 ( d S/$ U( d S/$ U R UUSSS9n/ nU( a= SU; a7 SU; a1 [ [ US US 5 5 nUR S S9S 9 US U nU( d S /$ U$ ! [10 a n[
S
U 35 S/s S nA$ S nAff = f)N)�Error� )zInput requiredr )z No candidates/parties providedr zThis text is about {}.F)�candidate_labels�hypothesis_template� multi_label�labels�scoresc � � U S $ )N� � )�items r �<lambda>�+ElectionPredictor.predict.<locals>.<lambda>' s � �D��G� T)�key�reverse)zNo specific predictionr zError during prediction: )zPrediction Errorr )r �strip�list�zip�sortr r )r r r r �results� predictions�
scored_labelsr s r �predict�ElectionPredictor.predict s� � ����"�#� #���!1�!1�!3�!3�+�,� ,�'�;�<� <� /��o�o��!9�$<� !� &� �G� �K��8�w�.�8�w�3F� $�S���):�G�H�<M�%N� O�
� �"�"�';�T�"�J� +�,<�_� =� � �7�8�8� � ��� /� �-�a�S�1� 2�-�.� .�� /�s$ �A!B# �!B# �#11C�-C�=C�C)r r N)zfacebook/bart-large-mnli)� ) �__name__�12__module__� __qualname__�__firstlineno__�strr r �intr �floatr5 �__static_attributes__r' r+ r r r sP � ��3� � /�#� /��c�� /�]`� /�im�ns�tw�y~�t~�n� jA� /� /r+ r N)� transformersr �typingr r r r r r' r+ r �<module>rB s �� !� )� )�,/� ,/r+