Spaces:

acecalisto3
/

GitBot

Runtime error

acecalisto3 commited on May 17

Commit

1d35b60

•

1 Parent(s): b1086b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,8 @@ from transformers import pipeline
 from sentence_transformers import SentenceTransformer, util
 import os
 import requests
-from transformers import AutoModelForCodeGeneration, AutoTokenizerForCodeGeneration
 class InferenceClient:
     def __init__(self):

 from sentence_transformers import SentenceTransformer, util
 import os
 import requests
+from transformers import AutoModelForCausalLM, AutoTokenizer
 class InferenceClient:
     def __init__(self):