nehalelkaref commited on
Commit
4568f61
·
verified ·
1 Parent(s): 7e85535

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from flask import Flask, jsonify, request, render_template
2
- from transformers import AutoAdapterModel, AutoTokenizer, TextClassificationPipeline
3
  import numpy
4
 
5
  tokenizer = AutoTokenizer.from_pretrained("UBC-NLP/MARBERT")
@@ -22,7 +22,6 @@ def home():
22
  @app.route('/classify', methods = ['POST'])
23
  def classify():
24
  text = request.form['comment']
25
- print(text)
26
 
27
  prediction = pipe(text)
28
  labels = {"LABEL_0":"GULF", "LABEL_1":"LEVANT","LABEL_2":"EGYPT"}
 
1
  from flask import Flask, jsonify, request, render_template
2
+ from adapters import AutoAdapterModel, AutoTokenizer, TextClassificationPipeline
3
  import numpy
4
 
5
  tokenizer = AutoTokenizer.from_pretrained("UBC-NLP/MARBERT")
 
22
  @app.route('/classify', methods = ['POST'])
23
  def classify():
24
  text = request.form['comment']
 
25
 
26
  prediction = pipe(text)
27
  labels = {"LABEL_0":"GULF", "LABEL_1":"LEVANT","LABEL_2":"EGYPT"}