Spaces:
Runtime error
Runtime error
File size: 707 Bytes
b38350f f9f50dd b38350f f9f50dd b38350f f9f50dd b38350f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from transformers import AutoTokenizer, AutoModelForCausalLM
from transformers import pipeline
import gradio as gr
#tokenizer = AutoTokenizer.from_pretrained("Imran1/gpt2-urdu-news")
#model = AutoModelForCausalLM.from_pretrained("Imran1/gpt2-urdu-news")
pipe = pipeline('text-generation',"Imran1/gpt2-urdu-news")
def text_generate(text):
generate_text = pipe(f"{text}", max_length = 200,top_k=2000, temperature=1.0, top_p=0.92)
return generate_text[0]['generated_text']
exmp=["درمدی قیمتوں میں","ایک فیصلے میں کہا کہ"]
gr.Interface(fn=text_generate, inputs="text", outputs="text", examples=exmp,title= "FINE TUNE GPT2 FOR URDU NEWS TEXT GENERATION").launch() |