Vineedhar commited on
Commit
46702b5
1 Parent(s): 6092885

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name, token=tok
8
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=token)
9
 
10
  # Define the pipeline with your model
11
- pipe = pipeline("question-answering", model=model, tokenizer=tokenizer)
12
 
13
  text = st.text_area("Enter some text:")
14
 
 
8
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=token)
9
 
10
  # Define the pipeline with your model
11
+ pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
12
 
13
  text = st.text_area("Enter some text:")
14