Redmind commited on
Commit
b87616e
·
verified ·
1 Parent(s): 85cec45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
-
3
  # Load the Open-Source LLM (e.g., BLOOM or Falcon)
4
  model_name = "tiiuae/falcon-7b-instruct" # Replace with your desired model
5
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
+ import gradio as gr
3
  # Load the Open-Source LLM (e.g., BLOOM or Falcon)
4
  model_name = "tiiuae/falcon-7b-instruct" # Replace with your desired model
5
  tokenizer = AutoTokenizer.from_pretrained(model_name)