krish10 commited on
Commit
b77d41f
·
verified ·
1 Parent(s): bafa892

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import os
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
7
- # Load model from secret
8
  model_name = os.environ.get("MODEL_ID")
9
 
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -77,7 +77,7 @@ Conclusion: Our study revealed a significant age-dependent acceleration in WMD v
77
  }
78
 
79
  with gr.Blocks() as demo:
80
- gr.Markdown("## 🧠 RCT Classifier Demonstration (Streaming, Stateless)")
81
 
82
  chatbot = gr.Chatbot(label="Model Response", type="tuples")
83
 
 
4
  import os
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
7
+ # Load model
8
  model_name = os.environ.get("MODEL_ID")
9
 
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
77
  }
78
 
79
  with gr.Blocks() as demo:
80
+ gr.Markdown("## 🧠 RCT Classifier Demonstration")
81
 
82
  chatbot = gr.Chatbot(label="Model Response", type="tuples")
83