Update app.py

app.py

@@ -7,10 +7,19 @@
 #----------------------------------------------------------------------------------------------------------------------------
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "DR-DRR/Model_001"
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+# Path to the model directory (assuming it's in the same directory as your script)
+model_directory = "./"
 
+# Load the model and tokenizer
+model = AutoModelForCausalLM.from_pretrained(model_directory)
+tokenizer = AutoTokenizer.from_pretrained(model_directory, trust_remote_code=True)
+
+# Now you can generate text as before
+# prompt = "What is a large language model?"
+# input_ids = tokenizer.encode(prompt, return_tensors="pt")
+
+# output = model.generate(input_ids, max_length=200, num_return_sequences=1)
+# generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
 # print(generated_text)
 
 # ---------------------------------------------------------------------------------------------------------------------------------------------------------------------