Update app.py

app.py

@@ -1,9 +1,13 @@
 import streamlit as st
 import transformers
 from transformers import T5Tokenizer, T5ForConditionalGeneration
-model_name = "t5-base"
-tokenizer = T5Tokenizer.from_pretrained(model_name)
-model = T5ForConditionalGeneration.from_pretrained(model_name)
+#model_name = "t5-base"
+#tokenizer = T5Tokenizer.from_pretrained(model_name)
+#model = T5ForConditionalGeneration.from_pretrained(model_name)
+model_name = "indonesia/gpt-2-small-indonesia"
+model = GPT2LMHeadModel.from_pretrained(model_name)
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+
 def generate_response(input_text):
     input_ids = tokenizer.encode(input_text, return_tensors='pt')
     outputs = model.generate(input_ids,