kmkarakaya's picture
second
92d0772
raw
history blame
1.44 kB
import tensorflow as tf
#!pip install transformers[sentencepiece] -q
from transformers import AutoTokenizer, TFGPT2LMHeadModel
#!pip install gradio -q
import gradio as gr
review_model = TFGPT2LMHeadModel.from_pretrained("kmkarakaya/turkishReviews-ds")
review_tokenizer = AutoTokenizer.from_pretrained("kmkarakaya/turkishReviews-ds")
def generate_review(prompt):
input_ids = review_tokenizer.encode(prompt, return_tensors='tf')
context_length = 40
output = review_model.generate(
input_ids,
do_sample=True,
max_length=context_length,
top_k=10,
no_repeat_ngram_size=2,
early_stopping=True
)
return(review_tokenizer.decode(output[0], skip_special_tokens=True))
demo = gr.Interface(fn=generate_review,
inputs="text",
outputs="text",
examples=[
"Bir hafta önce aldığım cep telefonu",
"Tatil için rezervasyon yaptırdım.",
"Henüz alalı bir"],
title="Review Generator",
description= """
Generate a review in Turkish by providing a prompt.
Generation takes 15 seconds on average.
Full Code is here: https://youtube.com/playlist?list=PLQflnv_s49v9d9w-L0S8XUXXdNks7vPBL
"""
)
demo.launch()