Update README.md

README.md

@@ -8,11 +8,11 @@ Tags:
 - T5
 - text-generation-inference
 ---
-Model Card for Easy_QG_Generator
+Model Card for Medium_QG_Generator
 
 Model Details
 
-Model Name: Easy_QG_Generator
+Model Name: Medium_QG_Generator
 
 Model Description: This model can generate MCQs from a given passage.
 
@@ -44,18 +44,18 @@ Usage
 Example Usage:
 
 Python
-easy_model = T5ForConditionalGeneration.from_pretrained("goenkalokesh/Easy_QG_Generator")
+medium_model = T5ForConditionalGeneration.from_pretrained("goenkalokesh/Medium_QG_Generator")
 
-easy_tokenizer = T5Tokenizer.from_pretrained("goenkalokesh/Easy_QG_Generator")
+medium_tokenizer = T5Tokenizer.from_pretrained("goenkalokesh/Medium_QG_Generator")
 
 --------------
 prompt ='''Your Prompt'''
 
 input_text = f"{prompt}: {passage.strip()}"
 
-input_ids = easy_tokenizer.encode(input_text, return_tensors='pt')
+input_ids = medium_tokenizer.encode(input_text, return_tensors='pt')
 
-outputs = easy_model.generate(
+outputs = medium_model.generate(
 
     input_ids, 
     
@@ -69,6 +69,6 @@ outputs = easy_model.generate(
     
 )
 
-output_text = easy_tokenizer.decode(outputs[0], skip_special_tokens=True)
+output_text = medium_tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 print(output_text)