Tafar commited on
Commit
3a64bb6
·
1 Parent(s): bb793ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -3
app.py CHANGED
@@ -1,7 +1,22 @@
1
  import gradio as gr
 
2
 
3
- def greet(Tafar):
4
- return "Hello " + name + "!!"
 
 
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 
 
 
 
 
 
 
 
 
 
 
7
  iface.launch()
 
1
  import gradio as gr
2
+ from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
+ # Load pre-trained GPT-3 model and tokenizer
5
+ model_name = "openai/gpt-3.5-turbo"
6
+ model = AutoModelForCausalLM.from_pretrained(model_name)
7
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
8
 
9
+ # Function to generate responses
10
+ def generate_response(prompt):
11
+ input_text = "You: " + prompt
12
+ input_ids = tokenizer.encode(input_text, return_tensors="pt")
13
+
14
+ # Generate response
15
+ response_ids = model.generate(input_ids, max_length=50, num_return_sequences=1)
16
+ response = tokenizer.decode(response_ids[0], skip_special_tokens=True)
17
+
18
+ return response
19
+
20
+ # Interface for Gradio
21
+ iface = gr.Interface(fn=generate_response, inputs="text", outputs="text", live=True, title="Glo - Your Chatbot")
22
  iface.launch()