Spaces:

brendon-ai
/

faq-huggingface-model

Running

brendon-ai commited on Jun 20

Commit

dd0d8dd

verified ·

1 Parent(s): 1cf63a4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 # Load tokenizer and model
 print("Loading tokenizer...")

 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+MODEL_NAME = "tiiuae/falcon-rw-1b" #"TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 # Load tokenizer and model
 print("Loading tokenizer...")