Update app.py

app.py

@@ -2,8 +2,8 @@ import torch
 import gradio as gr
 from transformers import XGLMTokenizer, XGLMForCausalLM
 
-tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-1.7B")
-model = XGLMForCausalLM.from_pretrained("facebook/xglm-1.7B")
+tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-2.9B")
+model = XGLMForCausalLM.from_pretrained("facebook/xglm-2.9B", low_cpu_mem_usage=True)
 
 def generate(text):
   inputs = tokenizer(text, return_tensors="pt")