Update app.py

app.py

@@ -23,9 +23,9 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 
-tokenizer = AutoTokenizer.from_pretrained("bigscience/bloom")
+tokenizer = AutoTokenizer.from_pretrained("bigscience/bloom-560m")
 
-model = AutoModelForCausalLM.from_pretrained("bigscience/bloom")
+model = AutoModelForCausalLM.from_pretrained("bigscience/bloom-560m")
 
 def get_result_with_bloom(text):
   result_length = 200