Update app.py
Browse files
app.py
CHANGED
|
@@ -105,11 +105,11 @@ tokenizer.push_to_hub("jonACE/llama-2-7b-chat_fine_tuned", token=hf_token)
|
|
| 105 |
|
| 106 |
|
| 107 |
# CHATBOT START
|
| 108 |
-
chatbot = pipeline("text-generation", model="jonACE/llama-2-7b-chat_fine_tuned")
|
| 109 |
|
| 110 |
-
def chatbot_response(prompt):
|
| 111 |
-
result = chatbot(prompt, max_length=100, do_sample=True, temperature=0.7)
|
| 112 |
-
return result[0]["generated_text"]
|
| 113 |
|
| 114 |
-
iface = gr.Interface(fn=chatbot_response, inputs="text", outputs="text")
|
| 115 |
-
iface.launch()
|
|
|
|
| 105 |
|
| 106 |
|
| 107 |
# CHATBOT START
|
| 108 |
+
# chatbot = pipeline("text-generation", model="jonACE/llama-2-7b-chat_fine_tuned")
|
| 109 |
|
| 110 |
+
#def chatbot_response(prompt):
|
| 111 |
+
# result = chatbot(prompt, max_length=100, do_sample=True, temperature=0.7)
|
| 112 |
+
# return result[0]["generated_text"]
|
| 113 |
|
| 114 |
+
#iface = gr.Interface(fn=chatbot_response, inputs="text", outputs="text")
|
| 115 |
+
#iface.launch()
|