Update app.py

app.py

@@ -4,9 +4,6 @@ import tokenizers
 import gradio as gr
 import re
 
-from PIL import Image
-
-
 def get_model(model_name, model_path='pytorch_model.bin'):
     tokenizer = transformers.GPT2Tokenizer.from_pretrained(model_name)
     model = transformers.OPTForCausalLM.from_pretrained(model_name)
@@ -33,7 +30,7 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, lengt
     return list(map(tokenizer.decode, out))[0]
 
 
-model, tokenizer = get_model('big-kek/NeuroSkeptic', 'big-kek/NeuroSkeptic')
+model, tokenizer = get_model('big-kek/NeuroSkeptic', 'OPT13b-skeptic.bin')
 
 example = 'Who is Bill Gates really?'
 demo = gr.Interface(