Spaces:

RohanHBTU
/

ChattyAI_prototype

Sleeping

RohanHBTU commited on Jun 19

Commit

4df6182

•

1 Parent(s): f6f0eae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 from flask import Flask, render_template, request, jsonify
 import os
-os.environ['TRANSFORMERS_HOME'] = '/rks/cache'
 #from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers import T5Tokenizer, T5ForConditionalGeneration

 from flask import Flask, render_template, request, jsonify
 import os
+os.environ['TRANSFORMERS_HOME'] = './.cache'
+os.environ['SENTENCE_TRANSFORMERS_HOME'] = './.cache'
 #from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers import T5Tokenizer, T5ForConditionalGeneration