Update app.py

app.py

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