Update README.md

README.md

@@ -77,7 +77,7 @@ pip install unsloth torch
 from unsloth import FastLanguageModel
 import torch
 
-class SPARQLQueryGenerator:
+class BiKECompoundNameExtractor:
     def __init__(self, model_name: str, max_seq_length: int = 32768, load_in_4bit: bool = True):
         self.model, self.tokenizer = FastLanguageModel.from_pretrained(
             model_name=model_name,
@@ -97,7 +97,7 @@ class SPARQLQueryGenerator:
             {"role": "user", "content": article_text}
         ]
 
-    def generate_query(self, article_text: str, temperature: float = 0.01, max_new_tokens: int = 1024) -> str:
+    def extract_compound_name(self, article_text: str, temperature: float = 0.01, max_new_tokens: int = 1024) -> str:
         si = "<|im_start|>assistant<|im_sep|>"
         sf = "<|im_end|>"
         messages = self.build_prompt(article_text)
@@ -117,9 +117,9 @@ class SPARQLQueryGenerator:
 
 # --- Using the model ---
 if __name__ == "__main__":
-    generator = SPARQLQueryGenerator(model_name="aksw/Bike-name")
+    extractor = BiKECompoundNameExtractor(model_name="aksw/Bike-name")
     text = "Title, Abstract, Introduction, Background, Method, Results, Conclusion, References."
-    list_names = generator.generate_query(text)
+    list_names = extractor.extract_compound_name(text)
     print(list_names)
 ```