Spaces:

roshnn24
/

Figr-DataScience-Assignment

Sleeping

roshnn24 commited on Jan 16

Commit

a07279d

verified ·

1 Parent(s): 3037245

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,6 +13,7 @@ from werkzeug.utils import secure_filename
 from huggingface_hub import InferenceClient
 from langchain.llms.base import LLM
 from typing import Optional, List, Any
 # Initialize Flask application
 app = Flask(__name__)

 from huggingface_hub import InferenceClient
 from langchain.llms.base import LLM
 from typing import Optional, List, Any
+from transformers import AutoTokenizer, AutoModelForCausalLM
 # Initialize Flask application
 app = Flask(__name__)