umairrrkhan commited on
Commit
1f8430f
·
verified ·
1 Parent(s): 717b6c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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" # Replace with your model
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 the AI generates!",
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