Configure application file

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM  # Mengimpor tokeni
 # 1. Load tokenizer dan model dari HuggingFace Hub
 model_name = "wahdie11/text2py2"           # Ganti dengan path ke model Anda di HuggingFace Hub
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name, ignore_mismatched_sizes=True)
 
 # 2. Definisikan fungsi untuk menghasilkan kode dari input teks
 def generate_code(prompt):

requirements.txt

@@ -1,3 +1,3 @@
-transformers>=4.0.0
-torch
+transformers>=4.31.0
+torch>=2.0.0
 gradio