ssirikon commited on
Commit
34fb2e3
1 Parent(s): d3c5c8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,7 +21,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
21
 
22
  # Create a pipeline for text generation
23
  generator = pipeline(
24
- task="text-generation",
25
  model=model,
26
  tokenizer=tokenizer,
27
  max_new_tokens=50, # Adjust as needed
@@ -45,4 +45,4 @@ demo = gr.Interface(
45
  description="Enter an email and let the model generate the subject for you!",
46
  )
47
 
48
- demo.launch()
 
21
 
22
  # Create a pipeline for text generation
23
  generator = pipeline(
24
+ task="summarization",
25
  model=model,
26
  tokenizer=tokenizer,
27
  max_new_tokens=50, # Adjust as needed
 
45
  description="Enter an email and let the model generate the subject for you!",
46
  )
47
 
48
+ demo.launch(debug=True)