update

app.py

@@ -63,11 +63,13 @@ def run_transformers(model_name, input_text, num_tokens):
     # if oauth_token is None : 
     #     return "Error : To Compare please login to your HF account and make sure you have access to the used Llama models"
     # Load the model and tokenizer dynamically if needed (commented out for performance)
-    if model_name=="TinyLlama/TinyLlama-1.1B-Chat-v1.0" : 
-        tokenizer = AutoTokenizer.from_pretrained('./models/tinyllama')
-        model = AutoModelForCausalLM.from_pretrained('./models/tinyllama')
+    # if model_name=="TinyLlama/TinyLlama-1.1B-Chat-v1.0" : 
+    print(input_text)
     if input_text is None : 
         return "Please provide an input text for the model", None
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    
     # Encode the input text
     input_ids = tokenizer.encode(input_text, return_tensors="pt")