Spaces:

RohanHBTU
/

ChattyAI_prototype

Sleeping

RohanHBTU commited on Jun 19

Commit

223f3a7

•

1 Parent(s): a07d0a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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