Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 os.system("pip install -q gradio torch transformers")
-os.system('pip install --no-deps xformers "trl<0.9.0" peft accelerate bitsandbytes')
 
 import gradio as gr
 import torch
@@ -8,7 +7,7 @@ import random
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')
-model = GPT2LMHeadModel.from_pretrained('DharGPT-Small')
+model = GPT2LMHeadModel.from_pretrained('RandomNameAnd6/DharGPT-Small')
 
 def generate_text(prompt):
   input_ids = tokenizer.encode(prompt, return_tensors="pt")