Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
 import spaces
 import torch
+import os
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-model_name = "krish10/Mistral_nemo_14B_16bit_RCT_classifier_1.0"
+# Get model name from secret
+model_name = os.environ.get("MODEL_ID")
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(