English
shivanircodel commited on
Commit
c0e11cb
·
verified ·
1 Parent(s): 79b523e

Create feedback.py

Browse files
Files changed (1) hide show
  1. feedback.py +18 -0
feedback.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import GPT3LMHeadModel, GPT3Tokenizer
2
+
3
+ # Load pre-trained model and tokenizer
4
+ model_name = "gpt3" # You could choose another model here
5
+ model = GPT2LMHeadModel.from_pretrained(model_name)
6
+ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
7
+
8
+ def generate_feedback(user_answer):
9
+ # Encode user input to model's input format
10
+ inputs = tokenizer.encode(user_answer, return_tensors='pt')
11
+
12
+ # Generate response from the model
13
+ outputs = model.generate(inputs, max_length=150, num_return_sequences=1, no_repeat_ngram_size=2)
14
+
15
+ # Decode the model's output to text
16
+ feedback = tokenizer.decode(outputs[0], skip_special_tokens=True)
17
+
18
+ return feedback