Spaces:

ConradLax
/

doc_classifier

Sleeping

ConradLax commited on Dec 18, 2023

Commit

aaeedf0

1 Parent(s): da3b513

test: render html template

Files changed (3) hide show

Dockerfile CHANGED Viewed

@@ -7,7 +7,6 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
-RUN python ./onLaunch.py
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt


7
8	COPY ./requirements.txt /code/requirements.txt
9

10
11	RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
12

main.py CHANGED Viewed

@@ -5,10 +5,11 @@ app = Flask(__name__)
 @app.route('/')
 def index():
-    sentiment_analysis = pipeline("sentiment-analysis")
-    result = sentiment_analysis("I absolutely love this product!")
-    return render_template('index.html', res=jsonify({"sentiment": result[0]["label"], "score": result[0]["score"]}))
 import torch
 from transformers import pipeline
@@ -19,8 +20,8 @@ from PIL import Image
-classifier_doctype_processor = DonutProcessor.from_pretrained("calumpianojericho/donutclassifier_acctdocs_by_doctype")
-classifier_doctype_model = VisionEncoderDecoderModel.from_pretrained("calumpianojericho/donutclassifier_acctdocs_by_doctype")
 # # Load the sentiment analysis model
 # sentiment_analysis = pipeline("sentiment-analysis")

 @app.route('/')
 def index():
+    # sentiment_analysis = pipeline("sentiment-analysis")
+    # result = sentiment_analysis("I absolutely love this product!")
+    return render_template('index.html', name="aaa");
+    # return render_template('index.html', res=jsonify({"sentiment": result[0]["label"], "score": result[0]["score"]}))
 import torch
 from transformers import pipeline
+# classifier_doctype_processor = DonutProcessor.from_pretrained("calumpianojericho/donutclassifier_acctdocs_by_doctype")
+# classifier_doctype_model = VisionEncoderDecoderModel.from_pretrained("calumpianojericho/donutclassifier_acctdocs_by_doctype")
 # # Load the sentiment analysis model
 # sentiment_analysis = pipeline("sentiment-analysis")

templates/index.html CHANGED Viewed

@@ -9,7 +9,7 @@
 </head>
 <body>
     <h1>Welcome!</h1>
-    <ul>
         {% for key, value in res.items() recursive %}
             <li>
             {% if value is string %}
@@ -20,7 +20,12 @@
             {% endif %}
             </li>
         {% endfor %}
-        </ul>
     <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.2/dist/js/bootstrap.bundle.min.js" integrity="sha384-C6RzsynM9kWDrMNeT87bh95OGNyZPhcTNXj1NW7RuBCsyN/o0jlpcV8Qyq46cDfL" crossorigin="anonymous"></script>
 </body>
 </html>

 </head>
 <body>
     <h1>Welcome!</h1>
+    <!-- <ul>
         {% for key, value in res.items() recursive %}
             <li>
             {% if value is string %}
             {% endif %}
             </li>
         {% endfor %}
+        </ul> -->
+        {% if name %}
+            <h1>Hello {{ name }}!</h1>
+        {% else %}
+            <h1>Hello, World!</h1>
+        {% endif %}
     <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.2/dist/js/bootstrap.bundle.min.js" integrity="sha384-C6RzsynM9kWDrMNeT87bh95OGNyZPhcTNXj1NW7RuBCsyN/o0jlpcV8Qyq46cDfL" crossorigin="anonymous"></script>
 </body>
 </html>