Update app.py

app.py

@@ -5,7 +5,8 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
 #https://huggingface.co/facebook/opt-1.3b
 #generator = pipeline('text-generation', model="microsoft/DialoGPT-medium")
 tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
-model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
+#model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
+model = GPT2LMHeadModel.from_pretrained('zmbfeng/FineTune-1')
 
 def create_response(input_str):
   #output_raw= generator(input_str)
@@ -23,6 +24,6 @@ fn=create_response,
 inputs="text",
 outputs="text",
 title="LLM",
-description="This interface generates a respsone"
+description="This interface generates a respsone (fine tuned)"
 )
 demo.launch()