lab 4

 from transformers import pipeline

import gensim.downloader as api
wv = api.load("glove-wiki-gigaword-50")
base_word = "technology"
sim = [w for w, _ in wv.most_similar(base_word, topn=3)]
print("Similar words:", sim)
orig_prompt = "Explain the impact of technology on society."
enriched_prompt = orig_prompt + " Also discuss " + ", ".join(sim) + "."
gen = pipeline("text-generation", model="gpt2")
print("\nOriginal prompt response:\n",
      gen(orig_prompt, max_length=80)[0]["generated_text"])
print("\nEnriched prompt response:\n",
      gen(enriched_prompt, max_length=100)[0]["generated_text"])

Comments

Popular posts from this blog

lab 6

lab 5

lab 1