update model to instruct

requirements.txt

@@ -1,5 +1,6 @@
 huggingface-hub
 transformers
+gradio
 
 
 # Linting and formatting

src/app.py

@@ -3,33 +3,34 @@ from transformers import pipeline
 import os
 from huggingface_hub import login
 
-login(token=os.getenv('HF_TOKEN'))
+login(token=os.getenv("HF_TOKEN"))
 
 
 def setup_pipeline():
-    # return pipeline(
-    #     "text-generation",
-    #     model="meta-llama/Llama-3.2-1B",  # Smaller model suitable for CPU
-    #     device=-1  # Force CPU
-    # )
-    return None
+    return pipeline(
+        "text-generation",
+        model="meta-llama/Llama-3.2-1B-instruct",  # Smaller model suitable for CPU
+        device=-1,  # Force CPU
+    )
+
 
 def generate_recipe(dish_name):
     if not dish_name:
         return "Please enter a dish name"
-    
+
     try:
-        
+
         prompt = f"""Create a recipe for {dish_name} including:
         - Ingredients with quantities
         - Steps to cook
         - Cultural background"""
-        
+
         result = generator(prompt, max_length=500, num_return_sequences=1)
-        return result[0]['generated_text']
+        return result[0]["generated_text"]
     except Exception as e:
         return f"Error: {str(e)}"
-    
+
+
 generator = setup_pipeline()
 
 demo = gr.Interface(
@@ -37,9 +38,8 @@ demo = gr.Interface(
     inputs=gr.Textbox(label="Enter dish name"),
     outputs=gr.Textbox(label="Generated Recipe", lines=20),
     title="RecipeGenie",
-    description="AI-powered recipe generator"
+    description="AI-powered recipe generator",
 )
 
 if __name__ == "__main__":
     demo.launch()
-

tests/test_app.py

@@ -2,33 +2,36 @@ import pytest
 from unittest.mock import Mock, patch
 from src.app import generate_recipe, setup_pipeline
 
-# @pytest.fixture
-# def mock_pipeline():
-#     with patch('src.app.pipeline') as mock:
-#         mock_generator = Mock()
-#         mock_generator.return_value = [{'generated_text': 'Test recipe output'}]
-#         mock.return_value = mock_generator
-#         yield mock
-
-# def test_empty_input():
-#     result = generate_recipe("")
-#     assert "Please enter a dish name" in result
-
-# def test_generate_recipe_success(mock_pipeline):
-#     result = generate_recipe("Pasta")
-#     assert isinstance(result, str)
-#     assert "Test recipe output" in result
-
-# def test_generate_recipe_exception():
-#     with patch('src.app.generator', side_effect=Exception("Test error")):
-#         result = generate_recipe("Pasta")
-#         assert "Error:" in result
-
-# def test_pipeline_creation():
-#     with patch('src.app.pipeline') as mock_pipeline:
-#         setup_pipeline()
-#         mock_pipeline.assert_called_once_with(
-#             "text-generation",
-#             model="meta-llama/Llama-3.2-1B",
-#             device=-1
-#         )
+
+@pytest.fixture
+def mock_pipeline():
+    with patch("src.app.pipeline") as mock:
+        mock_generator = Mock()
+        mock_generator.return_value = [{"generated_text": "Test recipe output"}]
+        mock.return_value = mock_generator
+        yield mock
+
+
+def test_empty_input():
+    result = generate_recipe("")
+    assert "Please enter a dish name" in result
+
+
+def test_generate_recipe_success(mock_pipeline):
+    result = generate_recipe("Pasta")
+    assert isinstance(result, str)
+    assert "Pasta" in result
+
+
+def test_generate_recipe_exception():
+    with patch("src.app.generator", side_effect=Exception("Test error")):
+        result = generate_recipe("Pasta")
+        assert "Error:" in result
+
+
+def test_pipeline_creation():
+    with patch("src.app.pipeline") as mock_pipeline:
+        setup_pipeline()
+        mock_pipeline.assert_called_once_with(
+            "text-generation", model="meta-llama/Llama-3.2-1B-instruct", device=-1
+        )