Update gen_ai.py

gen_ai.py

@@ -32,7 +32,7 @@ class traditional_model:
         predicted_class = torch.argmax(outputs.logits, dim=1).item()
         prediction = list(label_dict.keys())[predicted_class]
         return prediction
-class ResultB:
+class llm_model:
     def __init__(self, query):
         self.response = f"Result from Function B for query: {query}"