Update README.md

README.md

@@ -1,27 +0,0 @@
-import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer
-
-model = AutoModelForCausalLM.from_pretrained(
-                                              "ayucel/multilabel_classification"
-                                              device_map={"":0},
-                                              torch_dtype=torch.float16,
-)
-tokenizer = AutoTokenizer.from_pretrained("ayucel/multilabel_classification")
-
-prompt = "The phone met all my expectations, but I wish the shipping could have as well."
-
-pipe = pipeline("text-generation",
-                model=model,
-                tokenizer=tokenizer,
-                device_map={"":0},
-                max_length=250,
-                temperature=0.1,
-                top_p=1.0,
-                stop_sequence="}",
-                truncation=True
-                )
-
-result = pipe(f"<s>[INST] <<SYS>> Classify the text based on comments about the product and shipping as positive, negative, neutral, and None.<</SYS>> {prompt} [/INST]")
-print(result[0]['generated_text'])
-
-output="<s>[INST] <<SYS>> Classify the text based on comments about the product and shipping as positive, negative, neutral, and None.<</SYS>> The phone met all my expectations, but I wish the shipping could have as well. [/INST] {product: positive, shipping: negative}"