Update main.py

main.py

@@ -1,5 +1,5 @@
 import os
-os.environ['TRANSFORMERS_CACHE'] = '/tmp'
+os.environ['HF_HOME'] = '/tmp'
 from flask import Flask, render_template, request, jsonify
 from transformers import pipeline, AutoTokenizer