Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-def greet(name):
-    return "Hello " + name + "!!"
+tokenizer = AutoTokenizer.from_pretrained("PygmalionAI/pygmalion-6b")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+model = AutoModelForCausalLM.from_pretrained("PygmalionAI/pygmalion-6b")
+
+inputs = tokenizer("Hello", return_tensors="pt")
+outputs = model(**inputs, labels=inputs["input_ids"])
+loss = outputs.loss
+logits = outputs.logits