RohanHBTU commited on
Commit
43440d4
1 Parent(s): 8213fc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from flask import Flask, render_template, request, jsonify
2
  import os
3
- os.environ['SENTENCE_TRANSFORMERS_HOME'] = '/rks/cache'
4
  #from transformers import AutoModelForCausalLM, AutoTokenizer
5
  from transformers import T5Tokenizer, T5ForConditionalGeneration
6
 
 
1
  from flask import Flask, render_template, request, jsonify
2
  import os
3
+ os.environ['TRANSFORMERS_HOME'] = '/rks/cache'
4
  #from transformers import AutoModelForCausalLM, AutoTokenizer
5
  from transformers import T5Tokenizer, T5ForConditionalGeneration
6