Update app.py

app.py

@@ -1,13 +1,11 @@
 import gradio as gr
-from transformers import pipeline
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# tokenizer = AutoTokenizer.from_pretrained("Salesforce/xgen-7b-8k-base", trust_remote_code=True)
-# model = AutoModelForCausalLM.from_pretrained("Salesforce/xgen-7b-8k-base", torch_dtype=torch.bfloat16)
+tokenizer = AutoTokenizer.from_pretrained("Salesforce/xgen-7b-8k-base", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Salesforce/xgen-7b-8k-base", torch_dtype=torch.bfloat16)
 
 
-# pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
 def gentext(user_input="The world is"):
     inputs = tokenizer(user_input, return_tensors="pt")