TEST_6

app.py

@@ -5,7 +5,7 @@ from transformers import pipeline
 
 # Modelos
 chat_model_zephyr = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-chat_model_gemma = pipeline("text-generation", model="declare-lab/gemma-v2", max_new_tokens=256)
+chat_model_gpt2 = pipeline("text-generation", model="gpt2", max_new_tokens=100)
 
 # Similaridade
 similarity_model = SentenceTransformer("all-MiniLM-L6-v2")
@@ -25,8 +25,8 @@ def get_zephyr_response(question):
     return response.choices[0].message.content.strip()
 
 
-def get_gemma_response(question):
-    generated = chat_model_gemma(question)[0]["generated_text"]
+def get_gpt2_response(question):
+    generated = chat_model_gpt2(question)[0]["generated_text"]
     return generated.strip()
 
 
@@ -39,12 +39,12 @@ def compare_answers(answer1, answer2):
 
 def respond(question):
     answer_zephyr = get_zephyr_response(question)
-    answer_gemma = get_gemma_response(question)
-    similarity = compare_answers(answer_zephyr, answer_gemma)
+    answer_gpt2 = get_gpt2_response(question)
+    similarity = compare_answers(answer_zephyr, answer_gpt2)
 
     return (
         f"🧠 Zephyr-7b:\n{answer_zephyr}\n\n"
-        f"🤖 Gemma-v2:\n{answer_gemma}\n\n"
+        f"🤖 GPT-2:\n{answer_gpt2}\n\n"
         f"🔍 Similaridade Semântica: **{similarity}**"
     )