Update app.py

app.py

@@ -8,13 +8,13 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_id = 'K00B404/Merged_Beowolx-CodePro_Medusa2-7B-Mistral-I-v0-2'
+model_id = 'pip install accelerate'
 gr.load(f"models/{model_id}").launch()
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 model = AutoModelForCausalLM.from_pretrained(model_id)
 
-text = "Hello my name is"
+text = "TinyPixel/Llama-2-7B-bf16-sharded"
 inputs = tokenizer(text, return_tensors="pt")
 
 outputs = model.generate(**inputs, max_new_tokens=20)