Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ logger = logging.getLogger(__name__)
|
|
10 |
# Model and tokenizer setup
|
11 |
def setup_model_and_tokenizer():
|
12 |
logger.info("Loading model and tokenizer...")
|
13 |
-
model_name = "umairrrkhan/english-text-generation"
|
14 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
15 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
16 |
|
@@ -53,7 +53,7 @@ iface = gr.Interface(
|
|
53 |
inputs="text",
|
54 |
outputs="text",
|
55 |
title="AI Text Generation Chatbot",
|
56 |
-
description="Type a prompt and see what
|
57 |
examples=["Tell me a story about a robot.", "Write a poem about the moon."]
|
58 |
)
|
59 |
|
|
|
10 |
# Model and tokenizer setup
|
11 |
def setup_model_and_tokenizer():
|
12 |
logger.info("Loading model and tokenizer...")
|
13 |
+
model_name = "umairrrkhan/english-text-generation"
|
14 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
15 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
16 |
|
|
|
53 |
inputs="text",
|
54 |
outputs="text",
|
55 |
title="AI Text Generation Chatbot",
|
56 |
+
description="Lowkey curious? Type a prompt and see what Its generate! (190 character limit)",
|
57 |
examples=["Tell me a story about a robot.", "Write a poem about the moon."]
|
58 |
)
|
59 |
|