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
Post a Comment